einops>=0.4.1
torch>=1.10
torchvision
