torch>=2.0
einops>=0.7

[dev]
pytest
