accelerate
codetiming
datasets
dill
hydra-core
numpy>=2.0.0
pandas
peft
pyarrow>=19.0.0
pybind11
pylatexenc
ray[default]>=2.41.0
torchdata
tensordict!=0.9.0,<=0.10.0,>=0.8.0
transformers!=5.6.0,<5.11,>=5.5.3
wandb
packaging>=20.0
tensorboard

[geo]
mathruler
torchvision
qwen_vl_utils

[gpu]
liger-kernel
flash-attn

[math]
math-verify

[mcore]
megatron-bridge
mbridge

[prime]

[sglang]
tensordict!=0.9.0,<=0.10.0,>=0.8.0
sglang[openai,srt]==0.5.8
torch==2.9.1

[test]
pytest
pre-commit
py-spy
pytest-asyncio
pytest-rerunfailures

[trl]
trl<=0.9.6

[trtllm]
tensorrt-llm>=1.2.0rc6

[vllm]
tensordict!=0.9.0,<=0.10.0,>=0.8.0
vllm>=0.18.0
