numpy<2.3.0,>=2.0.0
datasets
omegaconf<3.0.0,>=2.3.0
fastapi
modelscope[framework]>=1.34.0
safetensors
peft<=0.19.0,>=0.11.0
transformers

[docs]
sphinx<6.0.0,>=5.3.0
docutils<0.17.0,>=0.16.0
myst_parser
recommonmark
sphinx-book-theme
sphinx-copybutton
sphinx-rtd-theme
sphinx_markdown_tables
sphinxcontrib-mermaid

[kernels]
kernels

[megatron]
megatron-core>=0.12.0
transformer-engine[pytorch]
mcore_bridge

[ray]
ray[serve]

[tinker]
tinker==0.14.0

[transformers]
accelerate
torch<3.0.0,>=2.6.0
torchvision

[vllm]
vllm>=0.11
