Cheng d01b83dfe7 Use nb::ndarray for checking arrays (#3283)

2026-03-25 10:44:54 +09:00

..

array.cpp

Fix cumulative operations when axis=None (#2653)

2025-10-08 15:25:38 -07:00

buffer.h

Double for lapack (#1904)

2025-02-25 11:39:36 -08:00

CMakeLists.txt

Fix python package install path in stubgen (#3009)

2026-01-19 09:34:02 +09:00

constants.cpp

Array api (#1289)

2024-07-26 10:40:49 -07:00

convert.cpp

Build and test python package on Windows CI (#3049)

2026-01-24 07:22:36 +09:00

convert.h

Support pickling array for bfloat16 (#2586)

2025-09-22 20:12:15 -07:00

cuda.cpp

Custom cuda kernel (#2517)

2025-08-20 17:20:22 -07:00

device.cpp

GPU discovery (#3055)

2026-01-26 09:54:13 -08:00

distributed.cpp

Thunderbolt RDMA communications backend (#2808)

2025-12-17 11:27:54 -08:00

export.cpp

Export: preserve Dtype state values in export callback arguments (#3145)

2026-02-19 08:07:28 -08:00

fast.cpp

Use lower-right causal mask alignment consistently (#2967)

2026-01-28 17:15:14 -08:00

fft.cpp

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

indexing.cpp

Use nb::ndarray for checking arrays (#3283)

2026-03-25 10:44:54 +09:00

indexing.h

Remove "using namespace mlx::core" in python/src (#1689)

2024-12-11 15:45:39 -08:00

linalg.cpp

typing: add type hints to mlx.core.array, linalg, distributed, and random (#2565)

2025-09-04 09:08:11 -07:00

load.cpp

Fix ref leak in mx.save/load with file like object (#3187)

2026-03-02 16:55:29 -08: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

GPU discovery (#3055)

2026-01-26 09:54:13 -08:00

mlx_func.cpp

Update nanobind pin to most recent version (#2896)

2025-12-11 06:07:36 -08:00

mlx_func.h

fix wraps compile (#2461)

2025-08-04 16:14:18 -07:00

mlx.cpp

only build for macos 14 and up (#2731)

2025-11-04 09:44:15 -08:00

ops.cpp

Add the bartlett function (#3155)

2026-03-03 11:40:54 -08:00

random.cpp

Create default random key lazily (#3278)

2026-03-19 20:22:52 +09:00

small_vector.h

Update nanobind pin to most recent version (#2896)

2025-12-11 06:07:36 -08:00

stream.cpp

Remove "using namespace mlx::core" in python/src (#1689)

2024-12-11 15:45:39 -08:00

transforms.cpp

Use thread local storage for frontend compile cache (#3280)

2026-03-20 07:44:45 +09:00

trees.cpp

Added support for pytree types that inherit from tuple and typing.namedtuple (#2845)

2025-12-04 11:06:45 -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

Build and test python package on Windows CI (#3049)

2026-01-24 07:22:36 +09:00

utils.h

Bump nanobind to 2.4 + fix (#1710)

2024-12-17 10:57:54 -08:00