mlx/examples/extensions
Awni Hannun 1156c84e86
Refactor common into cpu specific and truly common (#1817)
* refactor

* fix extension example

* fix no-cpu
2025-02-03 15:58:02 -08:00
..
axpby Refactor common into cpu specific and truly common (#1817) 2025-02-03 15:58:02 -08:00
mlx_sample_extensions Fix extensions (#1126) 2024-05-16 15:36:25 -07:00
bindings.cpp Remove "using namespace mlx::core" in benchmarks/examples (#1685) 2024-12-11 07:08:29 -08:00
CMakeLists.txt Bump nanobind to 2.4 + fix (#1710) 2024-12-17 10:57:54 -08:00
pyproject.toml Bump nanobind to 2.4 + fix (#1710) 2024-12-17 10:57:54 -08:00
README.md Rename block sparse (#1149) 2024-05-22 07:48:34 -07:00
requirements.txt Install mlx.dll into the same dir with python bindings on Windows (#1690) 2024-12-13 19:50:39 -08:00
setup.py bump nanobind + fix extension (#1344) 2024-08-21 16:05:07 -07:00
test.py Fix extensions (#1126) 2024-05-16 15:36:25 -07:00

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