torch>=2.9.1
transformers==5.9.0
safetensors>=0.5
huggingface-hub>=0.36.0
accelerate>=1.12.0
openai-harmony>=0.0.8
numpy>=2.2.0

[demos]
matplotlib>=3.10
jupyter>=1.1

[dev]
pytest>=8
ruff>=0.8

[train]
torchtune==0.6.1
torchao==0.13.0
datasets>=4.4
wandb>=0.23
tqdm>=4.66
omegaconf>=2.3

[vllm]
vllm==0.27.1
fastapi>=0.115
uvicorn>=0.30
pydantic>=2
