torch>=2.1.0
transformers<5,>=4.46
tokenizers>=0.19.0
safetensors>=0.4.0
numpy>=1.24.0
pyyaml>=6.0
tqdm>=4.66.0

[data]
datasets>=2.18.0

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

[eval]
lm-eval>=0.4.0

[train]
datasets>=2.18.0
tensorboard>=2.15.0

[wandb]
wandb>=0.16.0
