mlx/python/src
Anastasiia Filippova 9392fc3f88
NCCL backend (#2476)
2025-08-21 11:56:15 -07:00
..
array.cpp no segfault with uninitialized array.at (#2514) 2025-08-18 08:33:38 -07:00
buffer.h Double for lapack (#1904) 2025-02-25 11:39:36 -08:00
CMakeLists.txt Custom cuda kernel (#2517) 2025-08-20 17:20:22 -07:00
constants.cpp Array api (#1289) 2024-07-26 10:40:49 -07:00
convert.cpp Compile float64 functions on CPU (#2311) 2025-06-24 10:18:52 -07:00
convert.h Bump nanobind to 2.4 + fix (#1710) 2024-12-17 10:57:54 -08:00
cuda.cpp Custom cuda kernel (#2517) 2025-08-20 17:20:22 -07:00
device.cpp start cuda circle config (#2256) 2025-06-10 21:19:47 -07:00
distributed.cpp NCCL backend (#2476) 2025-08-21 11:56:15 -07:00
export.cpp Use SmallVector for shapes and strides (#2454) 2025-08-05 09:41:03 +09:00
fast.cpp Custom cuda kernel (#2517) 2025-08-20 17:20:22 -07:00
fft.cpp Use SmallVector for shapes and strides (#2454) 2025-08-05 09:41:03 +09:00
indexing.cpp Dynamic slicing (#1741) 2025-01-07 14:02:16 -08:00
indexing.h Remove "using namespace mlx::core" in python/src (#1689) 2024-12-11 15:45:39 -08:00
linalg.cpp Use SmallVector for shapes and strides (#2454) 2025-08-05 09:41:03 +09:00
load.cpp Use SmallVector for shapes and strides (#2454) 2025-08-05 09:41:03 +09:00
load.h Remove "using namespace mlx::core" in python/src (#1689) 2024-12-11 15:45:39 -08:00
memory.cpp move memory APIs into top level mlx.core (#1982) 2025-03-21 07:25:12 -07:00
metal.cpp Use SmallVector for shapes and strides (#2454) 2025-08-05 09:41:03 +09:00
mlx_func.cpp fix wraps compile (#2461) 2025-08-04 16:14:18 -07:00
mlx_func.h fix wraps compile (#2461) 2025-08-04 16:14:18 -07:00
mlx.cpp Custom cuda kernel (#2517) 2025-08-20 17:20:22 -07:00
ops.cpp Use SmallVector for shapes and strides (#2454) 2025-08-05 09:41:03 +09:00
random.cpp Use SmallVector for shapes and strides (#2454) 2025-08-05 09:41:03 +09:00
small_vector.h Use SmallVector for shapes and strides (#2454) 2025-08-05 09:41:03 +09:00
stream.cpp Remove "using namespace mlx::core" in python/src (#1689) 2024-12-11 15:45:39 -08:00
transforms.cpp Use SmallVector for shapes and strides (#2454) 2025-08-05 09:41:03 +09:00
trees.cpp Limit grad recursion depth by not recursing through non-grad inputs (#1764) 2025-01-14 14:33:18 -08:00
trees.h Limit grad recursion depth by not recursing through non-grad inputs (#1764) 2025-01-14 14:33:18 -08:00
utils.cpp iinfo and scalar overflow detection (#2009) 2025-03-27 19:54:56 -07:00
utils.h Bump nanobind to 2.4 + fix (#1710) 2024-12-17 10:57:54 -08:00