torch>=2.0.0
numpy
scikit-learn
safetensors

[dev]
pytest>=7.0.0
pytest-cov
ruff
black
mypy
