torch>=2.6.0
transformers>=4.40.0
datasets>=2.18.0
safetensors>=0.4.0
optuna>=3.6.0
numpy>=1.24.0
psutil>=5.9.0
tokenizers>=0.19.0
tqdm>=4.66.0
gradio<6,>=5.0

[cluster]
python-louvain>=0.16
networkx>=3.0

[dev]
pytest>=7.0
black>=24.0
ruff>=0.4.0
httpx>=0.27
fastapi==0.115.9
uvicorn==0.34.3

[mlx]
mlx>=0.32.4

[serve]
fastapi==0.115.9
uvicorn==0.34.3
