torch>=2.0
numpy
scipy
scikit-learn
matplotlib
pandas
pyyaml