mlx/python/src
Awni Hannun 2a980a76ce
Add stats and limit to common allocator and enable tests (#1988)
* add stats to common allocator and enable tests

* linux memory and default

* fix
2025-03-21 12:28:36 -07:00
..
array.cpp xor op on arrays (#1875) 2025-02-17 00:24:53 -08:00
buffer.h Double for lapack (#1904) 2025-02-25 11:39:36 -08:00
CMakeLists.txt move memory APIs into top level mlx.core (#1982) 2025-03-21 07:25:12 -07:00
constants.cpp Array api (#1289) 2024-07-26 10:40:49 -07:00
convert.cpp Double for lapack (#1904) 2025-02-25 11:39:36 -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 Ring docs (#1829) 2025-02-28 11:34:21 -08:00
export.cpp Cycle leak break (#1856) 2025-02-11 14:45:02 -08:00
fast.cpp Support fused masking in Attention (#1924) 2025-03-20 11:01:32 -07: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 Adds nuclear norm support (#1894) 2025-03-04 13:26:02 -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
memory.cpp move memory APIs into top level mlx.core (#1982) 2025-03-21 07:25:12 -07:00
metal.cpp Add stats and limit to common allocator and enable tests (#1988) 2025-03-21 12:28:36 -07:00
mlx_func.cpp Cycle leak break (#1856) 2025-02-11 14:45:02 -08:00
mlx_func.h Cycle leak break (#1856) 2025-02-11 14:45:02 -08:00
mlx.cpp move memory APIs into top level mlx.core (#1982) 2025-03-21 07:25:12 -07:00
ops.cpp Bitwise Inverse (#1862) 2025-02-13 08:44:14 -08:00
random.cpp Fix typo in randint docstring (#1932) 2025-03-05 21:48:00 -08:00
stream.cpp Remove "using namespace mlx::core" in python/src (#1689) 2024-12-11 15:45:39 -08:00
transforms.cpp fix grad with inplace updates (#1961) 2025-03-13 19:13:09 -07: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 fix int64 bug (#1860) 2025-02-12 19:23:46 -08:00
utils.h Bump nanobind to 2.4 + fix (#1710) 2024-12-17 10:57:54 -08:00