mlx/docs/src/python/optimizers
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
..
common_optimizers.rst Adding support for the Muon Optimizer (#1914) 2025-07-18 12:25:28 -07:00
optimizer.rst Support LR schedulers (#334) 2024-02-15 11:26:20 -08:00
schedulers.rst Add linear warmup and schedule joining for use with existing schedules (#721) 2024-02-26 07:28:48 -08:00