torch>=2.1.0
transformers>=4.38.0
peft>=0.8.0
datasets>=2.16.0
accelerate>=0.26.0
bitsandbytes>=0.42.0
trl>=0.7.0
sentencepiece>=0.1.99
protobuf>=3.20.0
optuna>=3.5.0
typer[all]>=0.9.0
rich>=13.0.0
pyyaml>=6.0
matplotlib>=3.8.0
jinja2>=3.1.0
psutil>=5.9.0
pynvml>=11.5.0
numpy>=1.24.0
unsloth
rouge-score
nltk
py-cpuinfo
tabulate
sentence-transformers
sacrebleu
scikit-learn

[dev]
pytest>=7.4.0
pytest-cov>=4.1.0
ruff>=0.2.0

[wandb]
wandb>=0.16.0
