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
pyzmq
accelerate
torch<3.0.0,>=2.6.0

[client]
textual>=1.0.0
plotext>=5.2.0
openai>=1.0.0
httpx>=0.25.0
tinker==0.16.1

[data]
py-data-juicer

[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

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

[rl]
vllm>=0.11
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
pytest
pytest-asyncio
