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