torch>=2.4.0
numpy>=1.26
einops>=0.7
safetensors>=0.4
pyyaml>=6.0
scipy>=1.13
rich>=13
transformers>=4.44
datasets>=2.19
huggingface-hub>=0.23
tqdm>=4.66
tensorboard>=2.17
wandb>=0.17

[dev]
pytest>=8
pytest-cov
mypy>=1.10
ruff>=0.4
