torch>=2.7.0
transformers==4.57.1
accelerate
einops
numpy
huggingface_hub

[dev]
pytest
pytest-xdist

[wandb]
wandb
