torch>=2.0
numpy>=1.20

[dev]
pytest>=7.0
scikit-learn>=1.0
