numpy
mlx
mlx-metal

[dev]
pytest
torch
torchvision

[gpu]
mlx
mlx-metal
