torch>=2.1.0
transformers>=4.36.0
numpy>=1.24.0

[all]
akv-cache[bench,dev,triton,vllm]

[bench]
accelerate
datasets
rich
matplotlib

[dev]
pytest>=7.0
pytest-cov
rich
ruff>=0.5

[triton]
triton>=2.1.0

[vllm]
vllm>=0.4.0
