mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-24 01:17:26 +08:00
MLX: An array framework for Apple silicon
| benchmarks | ||
| cmake | ||
| docs | ||
| examples | ||
| mlx | ||
| python | ||
| tests | ||
| .clang-format | ||
| .gitignore | ||
| .pre-commit-config.yaml | ||
| CMakeLists.txt | ||
| CODE_OF_CONDUCT.md | ||
| CONTRIBUTING.md | ||
| LICENSE | ||
| MANIFEST.in | ||
| mlx.pc.in | ||
| pyproject.toml | ||
| README.md | ||
| setup.py | ||
MLX
MLX is an array framework for machine learning specifically targeting Apple Silicon. MLX is designed with inspiration from Jax, PyTorch, ArrayFire.
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
Contributing
Check out the contribution guidelines for more information on contributing to MLX.