mlx/python/src
Awni Hannun c35f4d089a
start cuda circle config (#2256)
* rebase

* fix metal kernel linking issue on cuda

* start cuda circle config
2025-06-10 21:19:47 -07:00
..
array.cpp start cuda circle config (#2256) 2025-06-10 21:19:47 -07:00
buffer.h Double for lapack (#1904) 2025-02-25 11:39:36 -08:00
CMakeLists.txt fix linux linking error (#2248) 2025-06-06 10:41:51 -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 start cuda circle config (#2256) 2025-06-10 21:19:47 -07:00
distributed.cpp Min / max reductions (#2041) 2025-04-09 23:22:20 -07:00
export.cpp Use unordered map for kwargs in export/import (#2087) 2025-04-21 07:17:22 -07:00
fast.cpp Add new sdpa function overload (#2035) 2025-04-03 11:58:28 -07:00
fft.cpp add fftshift and ifftshift fft helpers (#2135) 2025-04-29 22:13:45 -07: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 non-symmetric eig and eigh (#2188) 2025-05-15 13:01:44 -07: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 Small typos in mx.metal deprecations (#2176) 2025-05-11 06:03:47 -07:00
mlx_func.cpp fix py gc edge case (#2079) 2025-04-18 12:46:53 -07: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 Clarify dimension notation in conv1d, conv2d, and conv3d docstrings (#2123) 2025-04-25 12:18:30 -07:00
random.cpp Add random normal distribution for complex numbers (#2182) 2025-05-13 22:43:45 -07:00
stream.cpp Remove "using namespace mlx::core" in python/src (#1689) 2024-12-11 15:45:39 -08:00
transforms.cpp fix py gc edge case (#2079) 2025-04-18 12:46:53 -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 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