mlx/python
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