mlx>=0.18.0
pydantic>=2.0
pyyaml>=6.0
numpy>=1.24.0
transformers>=4.35.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

[all]
mlx-forge[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

[wandb]
wandb>=0.16.0
