pandas>=2.2.2
numpy>=1.26.4
scipy>=1.13.1
scikit-learn>=1.6.0
iterative-stratification
torch>=2.5.1
bitsandbytes>=0.41.0
transformers[sentencepiece]<5.0.0,>=4.47.0
tokenizers>=0.21.0
peft
safetensors
datasets
seaborn
matplotlib
pydantic
openai
joblib

[dev]
pytest
ruff
black
