mlx/mlx
2025-04-14 16:37:36 -07:00
..
3rdparty jagrit's commit files 2023-11-29 10:52:08 -08:00
backend Gather mm new kernel and small refactoring (#2040) 2025-04-14 16:37:36 -07:00
distributed Fix the test and add custom min/max reductions for uncommon MPI types (#2060) 2025-04-10 17:01:17 -07:00
io redesign for faster cpu/gpu synch (#1869) 2025-03-06 19:23:38 -08:00
types Fp64 on the CPU (#1843) 2025-02-07 15:52:22 -08:00
allocator.cpp Add stats and limit to common allocator and enable tests (#1988) 2025-03-21 12:28:36 -07:00
allocator.h Add stats and limit to common allocator and enable tests (#1988) 2025-03-21 12:28:36 -07:00
array.cpp reduce binary size (#1952) 2025-03-11 06:30:44 -07:00
array.h reduce binary size (#1952) 2025-03-11 06:30:44 -07:00
CMakeLists.txt Do not define MLX_VERSION globally (#1966) 2025-03-18 07:12:40 -07:00
compile_impl.h Simplify removes no-ops from the tape (#1759) 2025-01-09 11:23:19 -08:00
compile.cpp Limit compile buffers (#1887) 2025-02-19 20:28:13 -08:00
compile.h fix function pointer (#1865) 2025-02-13 18:46:11 -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 fix double type promotion (#1901) 2025-02-25 06:00:53 -08:00
dtype.h Fp64 on the CPU (#1843) 2025-02-07 15:52:22 -08:00
einsum.cpp Einsum ellipsis (#1788) 2025-01-25 01:28:03 -08:00
einsum.h Einsum (#1269) 2024-07-25 09:36:44 -07:00
event.h Remove Event::Signal() (#2052) 2025-04-08 06:20:27 -07:00
export_impl.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
export.cpp LogCumSumExp (#2069) 2025-04-13 01:27:29 -07:00
export.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
fast_primitives.h Support fused masking in Attention (#1924) 2025-03-20 11:01:32 -07:00
fast.cpp Add float mask to sdpa vector (#2068) 2025-04-11 17:29:40 -07:00
fast.h Add new sdpa function overload (#2035) 2025-04-03 11:58:28 -07:00
fence.h redesign for faster cpu/gpu synch (#1869) 2025-03-06 19:23:38 -08:00
fft.cpp Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
fft.h Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
graph_utils.cpp Optionally specify names for arrays when exporting (#1749) 2025-01-06 13:07:46 -08:00
graph_utils.h Optionally specify names for arrays when exporting (#1749) 2025-01-06 13:07:46 -08:00
io.h Added missing unordered_map includes (#1635) 2024-12-02 07:03:03 -08:00
linalg.cpp Adds nuclear norm support (#1894) 2025-03-04 13:26:02 -08:00
linalg.h Adds nuclear norm support (#1894) 2025-03-04 13:26:02 -08:00
memory.h move memory APIs into top level mlx.core (#1982) 2025-03-21 07:25:12 -07:00
mlx.h move memory APIs into top level mlx.core (#1982) 2025-03-21 07:25:12 -07:00
ops.cpp Gather mm new kernel and small refactoring (#2040) 2025-04-14 16:37:36 -07:00
ops.h Gather mm new kernel and small refactoring (#2040) 2025-04-14 16:37:36 -07:00
primitives.cpp Gather mm new kernel and small refactoring (#2040) 2025-04-14 16:37:36 -07:00
primitives.h Gather mm new kernel and small refactoring (#2040) 2025-04-14 16:37:36 -07:00
random.cpp Adds nuclear norm support (#1894) 2025-03-04 13:26:02 -08:00
random.h Use int64 stride everywhere (#1671) 2024-12-09 11:09:02 -08:00
scheduler.cpp Do not join threads during process exit on Windows (#1738) 2025-03-25 06:33:08 -07:00
scheduler.h wait_for_one should wait for task to finish (#2047) 2025-04-05 20:05:16 -07:00
stream.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
threadpool.h Ring distributed backend (#1784) 2025-01-27 22:15:01 -08:00
transforms_impl.h no sdpa in grad (#2054) 2025-04-08 19:13:54 -07:00
transforms.cpp no sdpa in grad (#2054) 2025-04-08 19:13:54 -07:00
transforms.h Export / import functions to / from a file (#1642) 2024-12-24 11:19:13 -08:00
utils.cpp iinfo and scalar overflow detection (#2009) 2025-03-27 19:54:56 -07:00
utils.h iinfo and scalar overflow detection (#2009) 2025-03-27 19:54:56 -07:00
version.cpp Do not define MLX_VERSION globally (#1966) 2025-03-18 07:12:40 -07:00
version.h patch bump (#2043) 2025-04-03 13:15:18 -07:00