mlx>=0.31
numpy

[data]
tokenizers

[dev]
pytest

[wandb]
wandb
