mirror of
https://github.com/ml-explore/mlx.git
synced 2025-09-10 13:07:29 +08:00
818cda16bc
* Add a few LR schedulers * Move parents's constructor call to the top * Fix docstring * refactor optimizers into two files * add docs * nit * Fix Callable type annotation for python 3.8 --------- Co-authored-by: Awni Hannun <awni@apple.com> Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
24 lines
340 B
ReStructuredText
24 lines
340 B
ReStructuredText
Optimizer
|
|
=========
|
|
|
|
.. currentmodule:: mlx.optimizers
|
|
|
|
.. autoclass:: Optimizer
|
|
|
|
|
|
.. rubric:: Attributes
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
Optimizer.state
|
|
|
|
.. rubric:: Methods
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
Optimizer.apply_gradients
|
|
Optimizer.init
|
|
Optimizer.update
|