einops>=0.6
scipy
torch>=2
