torch>=2.0
numpy
scikit-learn
