pytorch-lightning<2,>=1.3.0
torch>=1.4
einops>=0.3.0
pandas>=1.0
scikit-learn>=0.24.2
hydra-core>=1.1.0
