Gökdeniz Gülmez deee214a95 Adding support for the Muon Optimizer (#1914) ...

* initial commit with workong optmimizer

* update ACKNOWLEDGMENTS.md

* nits and adding it to test

* nits

* G.astype(mx.bfloat16) to G.astype(G.dtype)

* G.ndim >= 2 to assert G.ndim == 2

* remove coments

* replace with  mx.addmm

* remove comments

* format

* nits

* match muon

* fix addmm

---------

Co-authored-by: Awni Hannun <awni@apple.com>

2025-07-18 12:25:28 -07:00

..

nn

Doc fix (#1615)

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

optimizers

Adding support for the Muon Optimizer (#1914)

2025-07-18 12:25:28 -07:00

array.rst

real and imag properties (#2189)

2025-05-15 18:17:50 -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

non-symmetric eig and eigh (#2188)

2025-05-15 13:01:44 -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