einops>=0.4
einops-exts
torch>=1.6
