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

..

array.cpp

Move to nanobind v2 (#1316)

2024-08-08 17:17:46 -07:00

buffer.h

Fix buffer protocol buffer size designation (#1010)

2024-04-19 06:06:13 -07:00

CMakeLists.txt

Comms (#1097)

2024-05-23 17:04:02 -07:00

constants.cpp

Array api (#1289)

2024-07-26 10:40:49 -07:00

convert.cpp

fix creating array from bf16 tensors in jax / torch (#1305)

2024-08-01 16:20:51 -07:00

convert.h

list based indexing (#1150)

2024-05-22 15:52:05 -07:00

device.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

distributed.cpp

Add docs for the distributed namespace (#1184)

2024-06-06 11:37:00 -07:00

fast.cpp

Custom Metal Kernels from Python (#1325)

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

fft.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

indexing.cpp

Fixes for large arrays with a few ops (#1299)

2024-07-30 17:18:39 -07:00

indexing.h

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

linalg.cpp

Custom Metal Kernels from Python (#1325)

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

load.cpp

Do not store iostream in shared_ptr (#872)

2024-03-22 06:54:45 -07:00

load.h

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

metal.cpp

Reset peak memory (#1074)

2024-05-03 17:12:51 -07:00

mlx.cpp

Comms (#1097)

2024-05-23 17:04:02 -07:00

ops.cpp

Add isfinite (#1318)

2024-08-13 14:49:28 -07:00

random.cpp

Einsum (#1269)

2024-07-25 09:36:44 -07:00

stream.cpp

Cpp docs (#1036)

2024-04-26 12:56:05 -07:00

transforms.cpp

docs fixes (#1263)

2024-07-11 15:59:07 -07:00

trees.cpp

Properly handle negative axes in python vmap (#944)

2024-04-02 18:07:23 -07:00

trees.h

Properly handle negative axes in python vmap (#944)

2024-04-02 18:07:23 -07:00

utils.cpp

fix conversion to array (#1070)

2024-05-06 16:02:49 -07:00

utils.h

fix conversion to array (#1070)

2024-05-06 16:02:49 -07:00