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

..

mlx

Fix optimizer reloading from checkpoint (#1329)

2024-08-15 07:33:23 -07:00

src

Custom Metal Kernels from Python (#1325)

2024-08-22 13:46:29 -07:00

tests

Custom Metal Kernels from Python (#1325)

2024-08-22 13:46:29 -07:00