accelerate
codetiming
datasets
dill
hydra-core
numpy
pandas
datasets
peft
pyarrow>=15.0.0
pybind11
pylatexenc
ray[default]>=2.10
tensordict<=0.6.2
torchdata
transformers
wandb

[geo]
mathruler

[gpu]
liger-kernel
flash-attn

[math]
math-verify

[prime]
pyext

[sglang]
tensordict<=0.6.2
sglang[all]==0.4.4.post4
torch-memory-saver>=0.0.5

[test]
pytest
yapf
py-spy

[vllm]
tensordict<=0.6.2
vllm<=0.8.2
