torch>=2.0.0
numpy<2.0,>=1.24.0
transformers>=4.36.0
accelerate>=0.20.0
sentencepiece>=0.1.99
pyyaml>=6.0

[cli]
rich>=13.0

[dev]
pytest>=7.0
ruff>=0.1.0
wandb>=0.15.0
rich>=13.0

[wandb]
wandb>=0.15.0
