einops>=0.7.0
torch>=1.10
torchvision

[test]
pytest
torch==2.4.0
torchvision==0.19.0
