numpy
mlx
mlx-metal
coremltools>=8.0

[coreml]
coremltools>=8.0

[dev]
pytest
torch
torchvision

[gpu]
mlx
mlx-metal
