mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-23 10:02:12 +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 | ||
array.rst | ||
data_types.rst | ||
devices_and_streams.rst | ||
distributed.rst | ||
export.rst | ||
fast.rst | ||
fft.rst | ||
linalg.rst | ||
memory_management.rst | ||
metal.rst | ||
nn.rst | ||
ops.rst | ||
optimizers.rst | ||
random.rst | ||
transforms.rst | ||
tree_utils.rst |