einops>=0.4
scipy
torch>=1.6
