torch>=1.6
einops>=0.6
transformers>=4.0
vit-pytorch>=0.40
routing-transformer>=1.6
