Back to Mlx

Optimizer

docs/src/python/optimizers/optimizer.rst

0.31.2339 B
Original Source

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