accelerate>=1.4.0
datasets>=3.0.0
transformers>=4.51.0

[bco]
scikit-learn
joblib

[deepspeed]
deepspeed>=0.14.4

[dev]
scikit-learn
joblib
deepspeed>=0.14.4
diffusers>=0.18.0
openai>=1.23.2
llm-blender>=0.0.2
liger-kernel>=0.5.9
peft>=0.8.0
bitsandbytes
parameterized
pytest-cov
pytest-rerunfailures
pytest-xdist
pytest
Pillow

[diffusers]
diffusers>=0.18.0

[judges]
openai>=1.23.2
llm-blender>=0.0.2

[liger]
liger-kernel>=0.5.9

[peft]
peft>=0.8.0

[quantization]
bitsandbytes

[scikit]
scikit-learn

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

[vllm]

[vllm:python_version < "3.13"]
vllm>=0.8.3
fastapi
pydantic
requests
uvicorn

[vlm]
Pillow
