mlx/python/mlx
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 allow parameters to be deleted (#2325) 2025-07-01 21:27:23 -07:00
optimizers Adding support for the Muon Optimizer (#1914) 2025-07-18 12:25:28 -07:00
__main__.py MLX in C++ example (#1736) 2025-01-02 19:09:04 -08:00
_os_warning.py Version bump and os error (#807) 2024-03-07 13:57:58 -08:00
_reprlib_fix.py override class function (#1418) 2024-09-16 13:21:04 -07:00
_stub_patterns.txt fix submodule stubs (#1492) 2024-10-15 16:23:37 -07:00
distributed_run.py Fix the launcher when ran locally (#2147) 2025-05-01 12:56:09 -07:00
extension.py Build with all cpu cores by default (#2336) 2025-07-07 06:06:45 -07:00
py.typed Add py.typed to support PEP-561 (type-hinting) for mlx (#588) 2024-01-31 12:05:42 -08:00
utils.py Add doc string for path (#1937) 2025-03-06 14:20:09 -08:00