torch>=2.0

[cuda]
triton

[dev]
pytest

[quantize]
transformers
datasets
safetensors
accelerate
sentencepiece
