mlx/examples/extensions
Awni Hannun 1ca616844b
Fix unintuitive metal kernel caching (#2242)
* Fix unintuitive metal kernel caching

* alternative solution
2025-06-06 20:08:15 -07:00
..
axpby
mlx_sample_extensions
bindings.cpp
CMakeLists.txt
pyproject.toml
README.md
requirements.txt
setup.py
test.py

Build

pip install -e .

For faster builds during development, you can also pre-install the requirements:

pip install -r requirements.txt

And then run:

python setup.py build_ext -j8 --inplace

Test

python test.py