einops<0.4,>=0.3
torch>=1.6
