einops>=0.4
einops-exts
torch>=1.6
vit-pytorch
