torch>=1.9
numpy
scikit-learn
