einops>=0.6.1
einops-exts
torch>=1.10
