mlx/python/src
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