torch>=2.0
numpy
matplotlib
scikit-learn

[dev]
pytest
