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	Add Primitive::name and remove Primitive::print (#2365)	2025-07-14 14:06:35 -07:00
cpu	Test with CUDA 12.2 (#2375)	2025-07-16 13:00:37 -07:00
cuda	Adding support for the Muon Optimizer (#1914)	2025-07-18 12:25:28 -07:00
gpu	Add contiguous_copy_gpu util for copying array (#2379)	2025-07-18 06:44:25 -07:00
metal	Add contiguous_copy_gpu util for copying array (#2379)	2025-07-18 06:44:25 -07:00
no_cpu	MoE backward improvements (#2335)	2025-07-07 17:59:53 -07:00
no_gpu	MoE backward improvements (#2335)	2025-07-07 17:59:53 -07:00