mlx/docs/src
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
..
_static Add Metal debug option and capture functions (#707) 2024-03-28 09:40:31 -07:00
_templates Quantize embedding (#994) 2024-04-15 16:42:10 -07:00
cpp Cpp docs (#1036) 2024-04-26 12:56:05 -07:00
dev Add Primitive::name and remove Primitive::print (#2365) 2025-07-14 14:06:35 -07:00
examples Fix docs latex build and nits (#1297) 2024-07-29 11:44:06 -07:00
python Adding support for the Muon Optimizer (#1914) 2025-07-18 12:25:28 -07:00
usage [CUDA] Fix back-end bugs and enable corresponding tests (#2296) 2025-06-16 08:45:40 -07:00
conf.py copyright in docs (#2214) 2025-05-21 17:13:04 -07:00
index.rst fix docs (#1991) 2025-03-21 19:58:53 -07:00
install.rst Install linux with mlx[cuda] and mlx[cpu] (#2356) 2025-07-14 17:17:33 -07:00