2023-11-30 02:52:08 +08:00
|
|
|
# MLX
|
|
|
|
|
|
|
|
MLX is an array framework for machine learning specifically targeting Apple
|
|
|
|
Silicon. MLX is designed with inspiration from Jax, PyTorch, ArrayFire.
|
|
|
|
|
2023-11-30 04:45:41 +08:00
|
|
|
[Documentation](https://ml-explore.github.io/mlx/build/html/index.html)
|
2023-11-30 02:52:08 +08:00
|
|
|
|
|
|
|
## Build
|
|
|
|
|
|
|
|
```
|
|
|
|
mkdir -p build && cd build
|
|
|
|
cmake .. && make -j
|
|
|
|
```
|
|
|
|
|
|
|
|
Run the C++ tests with `make test` (or `./tests/tests` for more detailed output).
|
|
|
|
|
|
|
|
### Python bidings
|
|
|
|
|
|
|
|
To install run:
|
|
|
|
|
|
|
|
`
|
|
|
|
env CMAKE_BUILD_PARALLEL_LEVEL="" pip install .
|
|
|
|
`
|
|
|
|
|
|
|
|
For developing use an editable install:
|
|
|
|
|
|
|
|
```
|
|
|
|
env CMAKE_BUILD_PARALLEL_LEVEL="" pip install -e .
|
|
|
|
```
|
|
|
|
|
|
|
|
To make sure the install is working run the tests with:
|
|
|
|
|
|
|
|
```
|
|
|
|
python -m unittest discover python/tests
|
|
|
|
```
|
|
|
|
|
|
|
|
|
2023-11-30 04:54:28 +08:00
|
|
|
## Contributing
|
2023-11-30 02:52:08 +08:00
|
|
|
|
2023-11-30 04:54:28 +08:00
|
|
|
Check out the [contribution guidelines](CONTRIBUTING.md) for more information
|
|
|
|
on contributing to MLX.
|