torch>=2.0

[bench]
vllm
datasets
huggingface_hub
jinja2
pyyaml
matplotlib
pandas

[cuda]
triton

[dev]
pytest

[hub]
jinja2
huggingface_hub
pyyaml

[quantize]
transformers
datasets
safetensors
accelerate
sentencepiece
