torch>=2.0.0
numpy>=1.20.0
scipy>=1.7.0
scikit-learn>=1.0.0
safetensors>=0.4.0
huggingface-hub>=0.20.0

[dev]
pytest>=7.0.0
matplotlib>=3.5.0
seaborn>=0.12.0
transformers>=4.30.0
