torch
numpy

[dev]
pytest
pygments
ruff
build
twine

[train]
transformers

[vllm]
vllm<0.20,>=0.19.1
