torch
triton
transformers>=4.45.0
tokenizers>=0.12.1
typing_extensions>=4.8.0
accelerate
datasets>=2.20
zstandard
huggingface_hub>=0.26.5

[cpu]
intel-extension-for-pytorch>=2.4.0

[dev]
black
mkdocstrings-python
mkdocs-material
griffe-typingdoc

[eval]
lm_eval==0.4.1
tabulate
protobuf
evaluate
scipy

[kernels]
autoawq-kernels
flash-attn>=2.2.0
