torch>=1.4.0
transformers>=4.40.0
accelerate
datasets
tyro>=0.5.11

[:platform_system != "Windows"]
numpy>=1.18.2

[:platform_system == "Windows"]
numpy<2

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

[deepspeed]
deepspeed>=0.14.4

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

[diffusers]
diffusers>=0.18.0

[liger]
liger-kernel>=0.2.1

[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
pytest-rerunfailures
llm-blender>=0.0.2
