accelerate
codetiming
datasets
dill
hydra-core
numpy
pandas
peft
pyarrow>=15.0.0
pybind11
pylatexenc
ray>=2.10
tensordict<0.6
torchdata
transformers
vllm<=0.6.3
wandb

[gpu]
liger-kernel
flash-attn

[prime]
pyext

[test]
pytest
yapf
py-spy
