torch>=2.0.0
numpy>=1.21.0

[all]
adaptive-k[dev,huggingface,vllm]

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
ruff>=0.1.0
mypy>=1.0.0

[huggingface]
transformers>=4.30.0
datasets>=2.0.0

[vllm]
vllm>=0.2.0
