mlx/mlx
2024-10-03 13:53:47 -07:00
..
3rdparty jagrit's commit files 2023-11-29 10:52:08 -08:00
backend faster cpu indexing (#1450) 2024-10-03 13:53:47 -07:00
distributed Data parallel helper (#1407) 2024-09-16 18:17:21 -07:00
io dtype is copy assignable (#1436) 2024-09-25 12:07:13 -07:00
types feat: metal formatting and pre-commit bump (#1038) 2024-04-30 07:18:09 -07:00
allocator.cpp Allow querying the allocator for the buffer size (#1404) 2024-09-11 21:02:16 -07:00
allocator.h Allow querying the allocator for the buffer size (#1404) 2024-09-11 21:02:16 -07:00
array.cpp Fix leak with multi-output primitives (#1274) 2024-07-23 06:34:18 -07:00
array.h Allow querying the allocator for the buffer size (#1404) 2024-09-11 21:02:16 -07:00
CMakeLists.txt Chore: add pre-commit hook for cmake (#1362) 2024-09-16 12:53:01 -07:00
compile_impl.h Ios compile (#784) 2024-03-04 20:02:26 -08:00
compile.cpp dtype is copy assignable (#1436) 2024-09-25 12:07:13 -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 minor fixes (#1194) 2024-06-12 22:06:49 -07:00
dtype.cpp dtype is copy assignable (#1436) 2024-09-25 12:07:13 -07:00
dtype.h dtype is copy assignable (#1436) 2024-09-25 12:07:13 -07:00
einsum.cpp Einsum (#1269) 2024-07-25 09:36:44 -07:00
einsum.h Einsum (#1269) 2024-07-25 09:36:44 -07:00
event.h minor fixes (#1194) 2024-06-12 22:06:49 -07:00
fast_primitives.h Some overhead reductions in mx.fast.metal_kernel (#1437) 2024-09-25 17:25:21 -07:00
fast.cpp Some overhead reductions in mx.fast.metal_kernel (#1437) 2024-09-25 17:25:21 -07:00
fast.h Simplifications for MLX C (#1396) 2024-09-06 19:16:50 -07:00
fft.cpp Feature complete Metal FFT (#1102) 2024-06-06 12:57:25 -07:00
fft.h Remove duplicate defines of StreamOrDevice and is_big_endian (#892) 2024-03-26 15:15:11 -07:00
graph_utils.cpp Remove unnecessary string copies (#891) 2024-03-28 13:14:59 -07:00
graph_utils.h Add missing && when forwarding args (#894) 2024-03-25 14:55:54 -07:00
io.h Add compiler flags to disable safetensors and gguf (#1098) 2024-05-09 17:39:44 -07:00
linalg.cpp feat: add cross_product (#1252) 2024-09-17 13:12:43 -07:00
linalg.h feat: add cross_product (#1252) 2024-09-17 13:12:43 -07:00
mlx.h Einsum (#1269) 2024-07-25 09:36:44 -07:00
ops.cpp allow take to work with integer index (#1440) 2024-09-26 15:58:03 -07:00
ops.h allow take to work with integer index (#1440) 2024-09-26 15:58:03 -07:00
primitives.cpp Put along axis + fixe for partition grad (#1430) 2024-09-23 10:03:38 -07:00
primitives.h Put along axis + fixe for partition grad (#1430) 2024-09-23 10:03:38 -07:00
random.cpp Einsum (#1269) 2024-07-25 09:36:44 -07:00
random.h Implement sampling from laplace distribution. (#1279) 2024-07-24 15:15:37 +02:00
scheduler.cpp Add synchronize function (#1006) 2024-04-22 08:25:46 -07:00
scheduler.h minor fixes (#1194) 2024-06-12 22:06:49 -07:00
stream.h Add synchronize function (#1006) 2024-04-22 08:25:46 -07:00
transforms_impl.h Custom transforms (#1246) 2024-07-10 18:00:01 -07:00
transforms.cpp Make eval just wait if all arrays are scheduled (#1368) 2024-08-27 17:01:22 -07:00
transforms.h Custom transforms (#1246) 2024-07-10 18:00:01 -07:00
utils.cpp dtype is copy assignable (#1436) 2024-09-25 12:07:13 -07:00
utils.h MPI ops in GPU stream for faster comms (#1356) 2024-08-26 15:12:50 -07:00