torch>=2.5
numpy
accelerate
pyyaml
huggingface_hub

[dev]
pytest

[logging]
wandb
