torch>=2.0.0
tqdm==4.66.4
click==8.1.7
safetensors~=0.4.3
accelerate~=0.30.1
pydantic==2.7.1
immutables==0.20
transformers>=4.42.3
huggingface_hub
peft
typing-extensions
sentencepiece
protobuf
scipy
datasets

[dev]
black~=24.4.2
isort~=5.13.2
pre-commit~=3.7.1

[evolve]
ray
cma
lm_eval
wandb

[test]
pytest~=8.2.1

[vllm]
vllm==0.3.2
lm_eval[vllm]
