torch>=1.8.1
scikit-learn
scipy
