einops>=0.6.1
torch>=1.91
numpy<2
