einops>=0.6.0
torch>=1.10
torchvision
