mlx/mlx
2024-03-13 16:22:03 -07:00
..
3rdparty jagrit's commit files 2023-11-29 10:52:08 -08:00
backend Make reshape faster for row_contiguous cases (#829) 2024-03-13 16:22:03 -07:00
io Remove unused variables (#706) 2024-02-18 12:50:10 -08:00
types Refactor CPU compile preamble (#708) 2024-02-19 06:12:53 -08:00
allocator.cpp Memory allocation (#292) 2024-01-02 11:59:19 -08:00
allocator.h Memory allocation (#292) 2024-01-02 11:59:19 -08:00
array.cpp Compile stride bug (#812) 2024-03-11 06:31:31 -07:00
array.h Compile stride bug (#812) 2024-03-11 06:31:31 -07:00
CMakeLists.txt Custom primitive + RoPE fat op (#676) 2024-02-14 14:04:25 -08:00
compile_impl.h Ios compile (#784) 2024-03-04 20:02:26 -08:00
compile.cpp NumberOfElements for shapeless compile and vmap fixes (#802) 2024-03-13 10:34:14 -07:00
compile.h Shapeless compilation for some graphs (#687) 2024-02-19 21:43:54 -08:00
device.cpp copyright + ack 2023-11-30 11:12:53 -08:00
device.h copyright + ack 2023-11-30 11:12:53 -08:00
dtype.cpp copyright + ack 2023-11-30 11:12:53 -08:00
dtype.h Pickle + dtype fix for numpy conversion (#763) 2024-03-02 06:09:29 -08:00
fast_primitives.h Fast Inference SDPA op (#735) 2024-03-04 21:06:11 -08:00
fast.cpp route to fallback for bfloat (#794) 2024-03-06 15:39:12 -08:00
fast.h Fast Inference SDPA op (#735) 2024-03-04 21:06:11 -08:00
fft.cpp Spelling (#342) 2024-01-01 21:08:17 -08:00
fft.h copyright + ack 2023-11-30 11:12:53 -08:00
graph_utils.cpp Kernel generation (#614) 2024-02-07 13:15:59 -08:00
graph_utils.h Kernel generation (#614) 2024-02-07 13:15:59 -08:00
io.h Metadata support for safetensors (#639) 2024-02-08 19:33:15 -08:00
linalg.cpp Add SVD primitive (#809) 2024-03-12 12:30:11 -07:00
linalg.h Add SVD primitive (#809) 2024-03-12 12:30:11 -07:00
mlx.h Custom primitive + RoPE fat op (#676) 2024-02-14 14:04:25 -08:00
ops.cpp NumberOfElements for shapeless compile and vmap fixes (#802) 2024-03-13 10:34:14 -07:00
ops.h NumberOfElements for shapeless compile and vmap fixes (#802) 2024-03-13 10:34:14 -07:00
primitives.cpp NumberOfElements for shapeless compile and vmap fixes (#802) 2024-03-13 10:34:14 -07:00
primitives.h NumberOfElements for shapeless compile and vmap fixes (#802) 2024-03-13 10:34:14 -07:00
random.cpp Add loc and scale to random.normal (#638) 2024-02-07 11:49:59 -08:00
random.h Add loc and scale to random.normal (#638) 2024-02-07 11:49:59 -08:00
scheduler.cpp copyright + ack 2023-11-30 11:12:53 -08:00
scheduler.h move all ObjC (via metal-cpp) interaction until post static initializers (#370) 2024-01-04 16:12:00 -08:00
stream.h copyright + ack 2023-11-30 11:12:53 -08:00
transforms_impl.h Ios compile (#784) 2024-03-04 20:02:26 -08:00
transforms.cpp NumberOfElements for shapeless compile and vmap fixes (#802) 2024-03-13 10:34:14 -07:00
transforms.h Compile primitive (#571) 2024-02-05 06:51:22 -08:00
utils.cpp Adds device context manager (#679) 2024-02-14 14:14:58 -08:00
utils.h CPU compile (#691) 2024-02-17 06:54:32 -08:00