mlx/docs/src/dev
Alex Barron 0fd2a1f4b0
Custom Metal Kernels from Python (#1325)
* start

* simple kernels working

* restructure

* inverse example working

* docs + fixes

* missing file

* fix imports

* address comments

* add docs + fix test

* Review comments + refactor to a single function

* update docs

* remove hashing

* fix contig bug in test

* back to a class

* trailing whitespace

* fix tests

* match c++ and python apis

* add link + make args kw_only
2024-08-22 13:46:29 -07:00
..
custom_metal_kernels.rst Custom Metal Kernels from Python (#1325) 2024-08-22 13:46:29 -07:00
extensions.rst Add gemv masked to JIT plus some fixes (#1310) 2024-08-07 13:38:07 -07:00
metal_debugger.rst Add synchronize function (#1006) 2024-04-22 08:25:46 -07:00