mlx>=0.18.0
pydantic>=2.0
pyyaml>=6.0
numpy>=1.24.0
transformers<5.0,>=4.35.0
jinja2>=3.0
safetensors>=0.4.0
huggingface-hub>=0.20.0
datasets>=2.16.0
fastapi>=0.104.0
uvicorn[standard]>=0.24.0
websockets>=12.0
sse-starlette>=1.6.0

[all]
mlx-forge[docs,vision,wandb]

[dev]
pytest>=7.0
pytest-timeout>=2.0
pytest-asyncio>=0.23.0
pytest-cov>=4.1.0
httpx>=0.25.0
ruff>=0.4.0

[dev:python_version < "3.11"]
tomli>=2.0

[docs]
mkdocs>=1.5
mkdocs-material>=9.0

[vision]
mlx-vlm>=0.4.0

[wandb]
wandb>=0.16.0
