mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-22 01:21:14 +08:00
![]() * 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> |
||
---|---|---|
.. | ||
nn | ||
optimizers | ||
__main__.py | ||
_os_warning.py | ||
_reprlib_fix.py | ||
_stub_patterns.txt | ||
distributed_run.py | ||
extension.py | ||
py.typed | ||
utils.py |