torch>=2.0.0
transformers>=4.33.0
datasets>=2.14.0
tokenizers>=0.15.0
numpy>=1.24.0
tqdm>=4.65.0

[dev]
pytest>=7.0
black>=23.0
ruff>=0.1.0
mypy>=1.0

[infer]
vllm>=0.5.0
sglang>=0.4.0
lmdeploy>=0.5.0

[train]
accelerate>=0.20.0
deepspeed>=0.10.0
