einops>=0.8.2
torch>=2.5
torchvision

[test]
pytest
torch==2.5.0
torchvision==0.20.0
