Awni Hannun 33421c1dd3 Limit grad recursion depth by not recursing through non-grad inputs (#1764) ...

* limit grad recursion depth

* add grad of module test

2025-01-14 14:33:18 -08:00

..

array.cpp

Add mx.finfo and use it when making causal mask (#1726)

2024-12-19 14:52:41 -08:00

buffer.h

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

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

CMakeLists.txt

Export / import functions to / from a file (#1642)

2024-12-24 11:19:13 -08:00

constants.cpp

Array api (#1289)

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

convert.cpp

Export / import functions to / from a file (#1642)

2024-12-24 11:19:13 -08:00

convert.h

Bump nanobind to 2.4 + fix (#1710)

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

device.cpp

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

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

distributed.cpp

mpi send use input as output (#1750)

2025-01-06 06:08:43 -08:00

export.cpp

Optionally specify names for arrays when exporting (#1749)

2025-01-06 13:07:46 -08:00

fast.cpp

Allow boolean mask in sdpa (#1753)

2025-01-06 16:57:07 -08:00

fft.cpp

More shape type (#1705)

2024-12-19 08:08:20 -08: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

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

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

load.cpp

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

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

load.h

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

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

metal.cpp

track resource limit and throw if exceeded (#1718)

2024-12-18 18:45:58 -08:00

mlx.cpp

Export / import functions to / from a file (#1642)

2024-12-24 11:19:13 -08:00

ops.cpp

Fix broadcast_arrays python sig (#1763)

2025-01-10 12:33:26 -08:00

random.cpp

More shape type (#1705)

2024-12-19 08:08:20 -08:00

stream.cpp

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

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

transforms.cpp

Limit grad recursion depth by not recursing through non-grad inputs (#1764)

2025-01-14 14:33:18 -08: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

Bump nanobind to 2.4 + fix (#1710)

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

utils.h

Bump nanobind to 2.4 + fix (#1710)

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