pytorch_lightning>=1.1.3
scipy
scikit-learn
