torch>=1.4.0
transformers>=4.31.0
numpy<2.0.0,>=1.18.2
accelerate
datasets
tyro>=0.5.11
peft
wandb
deepspeed
stanza
nltk
scipy

[benchmark]
wandb
ghapi
openrlbenchmark==0.2.1a5
requests
deepspeed

[deepspeed]
deepspeed>=0.9.5

[dev]
parameterized
peft>=0.8.0
pytest
pytest-xdist
pytest-cov
scikit-learn
Pillow
diffusers>=0.18.0
deepspeed>=0.9.5
wandb
ghapi
openrlbenchmark==0.2.1a5
requests
deepspeed
bitsandbytes<=0.41.1
openai>=1.23.2
huggingface_hub>=0.22.2
llm-blender>=0.0.2

[diffusers]
diffusers>=0.18.0

[llm_judge]
openai>=1.23.2
huggingface_hub>=0.22.2
llm-blender>=0.0.2

[peft]
peft>=0.8.0

[quantization]
bitsandbytes<=0.41.1

[test]
parameterized
peft>=0.8.0
pytest
pytest-xdist
pytest-cov
scikit-learn
Pillow
