npm

mlx

Array framework for Apple silicon machine learning

brewmacoslinux

Try with needOr install directly

Source

About

Array framework for Apple silicon

Commands

mlx

Examples

Create and manipulate arrays with MLX$ python3 -c "import mlx.core as mx; a = mx.array([1, 2, 3]); print(a)"

Train a simple neural network model$ python3 -c "import mlx.nn as nn; model = nn.Sequential(nn.Linear(10, 5), nn.ReLU(), nn.Linear(5, 1)); print(model)"

Perform efficient GPU computation on Apple silicon$ python3 -c "import mlx.core as mx; x = mx.random.normal((1000, 1000)); y = mx.matmul(x, x.T); print(y.shape)"