einops>=0.3
pytorch-lightning==0.8.5
torch>=1.6
