accelerate>=0.26.0
evaluate
datasets>=2.16.0
evaluate>=0.4.0
jsonlines
numexpr
peft>=0.2.0
pybind11>=2.6.2
pytablewriter
rouge-score>=0.0.4
sacrebleu>=1.5.0
scikit-learn>=0.24.1
sqlitedict
torch>=1.8
tqdm-multiprocess
transformers>=4.1
zstandard
dill
word2number
more_itertools

[all]
lm_eval[anthropic]
lm_eval[dev]
lm_eval[deepsparse]
lm_eval[gptq]
lm_eval[hf_transfer]
lm_eval[ifeval]
lm_eval[mamba]
lm_eval[math]
lm_eval[multilingual]
lm_eval[openai]
lm_eval[promptsource]
lm_eval[sentencepiece]
lm_eval[sparseml]
lm_eval[testing]
lm_eval[vllm]
lm_eval[zeno]
lm_eval[wandb]

[api]
requests
aiohttp
tenacity
tqdm
tiktoken

[deepsparse]
deepsparse-nightly[llm]>=1.8.0.20240404

[dev]
pytest
pytest-cov
pytest-xdist
pre-commit
mypy

[gptq]
auto-gptq[triton]>=0.6.0

[hf_transfer]
hf_transfer

[ifeval]
langdetect
immutabledict

[mamba]
mamba_ssm
causal-conv1d==1.0.2

[math]
sympy>=1.12
antlr4-python3-runtime==4.11

[multilingual]
nagisa>=0.2.7
jieba>=0.42.1
pycountry

[neuronx]
optimum[neuronx]

[optimum]
optimum[openvino]

[promptsource]
promptsource>=0.2.3

[sentencepiece]
sentencepiece>=0.1.98

[sparseml]
sparseml-nightly[llm]>=1.8.0.20240404

[testing]
pytest
pytest-cov
pytest-xdist

[vllm]
vllm>=0.4.2

[wandb]
wandb>=0.16.3
pandas
numpy

[zeno]
pandas
zeno-client
