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

..

nn

Doc fix (#1615)

2024-11-22 11:12:25 -08:00

optimizers

Fix stubgen (#2065)

2025-04-11 12:02:54 -07:00

array.rst

LogCumSumExp (#2069)

2025-04-13 01:27:29 -07:00

data_types.rst

Fp64 on the CPU (#1843)

2025-02-07 15:52:22 -08:00

devices_and_streams.rst

Add synchronize function (#1006)

2024-04-22 08:25:46 -07:00

distributed.rst

Adds send/recv ops in distributed (#1366)

2024-08-26 23:01:37 -07:00

export.rst

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

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

fast.rst

3 and 6 bit quantization (#1613)

2024-11-22 10:22:13 -08:00

fft.rst

add fftshift and ifftshift fft helpers (#2135)

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

linalg.rst

add pinv to doc (#2020)

2025-03-30 15:54:18 -07:00

memory_management.rst

fix docs (#1991)

2025-03-21 19:58:53 -07:00

metal.rst

fix docs (#1991)

2025-03-21 19:58:53 -07:00

nn.rst

Ring docs (#1829)

2025-02-28 11:34:21 -08:00

ops.rst

LogCumSumExp (#2069)

2025-04-13 01:27:29 -07:00

optimizers.rst

minor fix in optimizer + docs (#1264)

2024-07-12 12:18:02 -07:00

random.rst

Add mx.random.permutation (#1471)

2024-10-08 19:42:19 -07:00

transforms.rst

Add missing funcs to docs (#2021)

2025-03-30 18:29:33 -07:00

tree_utils.rst

feat: implement clip_grad_norm (#1043)

2024-05-03 09:07:02 -07:00