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