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
typer>=0.9.0

[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]

[server]
redis>=5.0
psutil>=5.9.0
pynvml>=11.0.0
opentelemetry-api
opentelemetry-sdk
opentelemetry-exporter-otlp
opentelemetry-instrumentation-logging

[test]
hypothesis>=6.0

[tinker]
tinker==0.14.0

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

[vllm]
vllm>=0.11
