torch>=2.0.0
einops>=0.6.0

[dev]
pytest>=7.4.0
pytest-cov>=4.1.0
ruff>=0.1.0
mypy>=1.5.0
pre-commit>=3.4.0
types-setuptools
