zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 17:31:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
0fd2a1f4b0
mlx/python/src
History
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