torch>=1.4.0
transformers>=4.31.0
numpy>=1.18.2
accelerate
datasets
tyro>=0.5.11

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

[deepspeed]
deepspeed>=0.9.5

[dev]
parameterized
pytest
pytest-xdist
accelerate
pytest-cov
peft>=0.4.0
diffusers>=0.18.0
deepspeed>=0.9.5
wandb
ghapi
openrlbenchmark==0.2.1a5
requests
deepspeed
bitsandbytes<=0.41.1

[diffusers]
diffusers>=0.18.0

[peft]
peft>=0.4.0

[quantization]
bitsandbytes<=0.41.1

[test]
parameterized
pytest
pytest-xdist
accelerate
pytest-cov
