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

..

mlx

Adding support for the Muon Optimizer (#1914)

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

scripts

Install linux with mlx[cuda] and mlx[cpu] (#2356)

2025-07-14 17:17:33 -07:00

src

Add Primitive::name and remove Primitive::print (#2365)

2025-07-14 14:06:35 -07:00

tests

Adding support for the Muon Optimizer (#1914)

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