torch>=2.1

[all]
turbo-attn[codebook,eval,flash-attn,flashinfer,triton,vllm]

[codebook]
scipy>=1.10

[dev]
pytest>=7.0
scipy>=1.10
triton>=3.0

[eval]
lm-eval>=0.4.5
ray
datasets
langdetect
immutabledict
nltk
sacrebleu
absl-py

[flash-attn]
flash-attn>=2.5

[flashinfer]
flashinfer>=0.6

[triton]
triton>=3.0

[vllm]
vllm>=0.19
