Aashiq Dheeraj bb6565ef14 add fftshift and ifftshift fft helpers (#2135) ...

* add fftshift and ifftshift fft helpers

* address comments

* axes have to be iterable

* fix fp error in roll + add test

---------

Co-authored-by: Aashiq Dheeraj <aashiq@aashiq-mbp-m4.local>

2025-04-29 22:13:45 -07:00

..

array.cpp

fix pinv (#2110)

2025-04-23 13:08:28 -07: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

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

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

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

Fix typos (#2136)

2025-04-29 07:26:05 -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