mlx>=0.5.0
mlx-lm>=0.10.0
numpy>=1.24.0
transformers>=4.35.0
pyyaml>=6.0
tqdm>=4.65.0

[all]
wandb>=0.15.0
scikit-learn>=1.3.0

[dev]
pytest>=7.0
black>=23.0
isort>=5.12

[sklearn]
scikit-learn>=1.3.0

[wandb]
wandb>=0.15.0
