mlx>=0.31.0
mlx-lm>=0.31.0
mlx-vlm>=0.6.0
numpy

[bench]
datasets

[dev]
pytest
torch
numpy
datasets
