torch>=2.1.2
numpy>=1.26.2
scipy>=1.11.4
protobuf>=5.27.2
pre-commit>=3.7.1
ipykernel>=6.29.5
genlm-backend>=0.1.0a1

[collie]
collie-bench>=0.1.0
nltk>=3.8.1
dill>=0.3.8
evaluate>=0.4.2

[dev]
pytest
pytest-benchmark
pytest-cov
pre-commit>=3.6.0
ruff>=0.9.9
jupyterlab>=4.0.9
ipywidgets>=8.1.1
matplotlib>=3.9.1
seaborn>=0.13.2

[examples]
nltk>=3.8.1

[mlx]
genlm-backend[mlx]>=0.1.7

[vllm]
vllm>=0.6.6

[yelp]
yake>=0.4.8
datasets>=2.20.0
