mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-14 00:39:06 +08:00
deee214a95
* 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>
23 lines
293 B
ReStructuredText
23 lines
293 B
ReStructuredText
.. _common_optimizers:
|
|
|
|
Common Optimizers
|
|
=================
|
|
|
|
.. currentmodule:: mlx.optimizers
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
:template: optimizers-template.rst
|
|
|
|
SGD
|
|
RMSprop
|
|
Adagrad
|
|
Adafactor
|
|
AdaDelta
|
|
Adam
|
|
AdamW
|
|
Adamax
|
|
Lion
|
|
MultiOptimizer
|
|
Muon
|