numpy>=1.20
pyyaml>=6

[deepspeed]
deepspeed==0.18.9

[dev]
pytest
ruff

[distributed]
ray[default]>=2.40.0

[full]
numpy==2.3.2
scipy==1.17.1
pandas==3.0.2
scikit-learn==1.8.0
joblib==1.5.3
networkx==3.5
numba==0.65.0
matplotlib==3.10.5
PyYAML==6.0.2
hydra-core==1.3.2
omegaconf==2.3.0
easydict==1.13
rich==15.0.0
tqdm==4.67.3
requests==2.33.1
httpx==0.28.1
pydantic==2.13.0
psutil==7.0.0
regex==2026.4.4
ftfy==6.3.1
huggingface-hub==1.10.2
openai==2.31.0
anthropic>=0.40.0
fastapi>=0.115.0
uvicorn[standard]>=0.32.0
shortuuid>=1.0.0
nano-vectordb==0.0.4.3
gdown==5.2.0
einops==0.8.2

[gcp]
google-cloud-aiplatform>=1.38.0

[llm]
transformers==5.5.4
tokenizers==0.22.2
safetensors==0.7.0
accelerate==1.13.0
bitsandbytes==0.49.2
peft==0.19.0
datasets==4.8.4
sentence-transformers==5.4.1
fschat>=0.2.36

[serve]
gradio>=5.0.0

[torch]
torch==2.8.0
torch-geometric>=2.3.0
torch-scatter==2.1.2
torch-sparse==0.6.18
torch-cluster==1.6.3
torch-spline-conv==1.2.2
ogb==1.3.6
geoopt==0.5.1
deeprobust>=0.2.4
scipy>=1.9.0
transformers>=4.36.0
joblib>=1.3.0
scikit-learn>=1.3.0
ftfy>=6.1.0
regex>=2023.0.0
accelerate>=0.26.0
fschat>=0.2.36

[trackers]
wandb>=0.19.0
swanlab<0.8,>=0.7.11
