docs/src/python/optimizers/optimizer.rst

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
Compile with capture (#629) * Simple kernel generation * Remove the generate kernel from graph_utils * fix multi-output with compile * fuse with stopgrad * v1 input, output capture in compile * cleanup tree update with visitor update * nit * remove todo * state for model, optional explicit init and more pure optimizer steps * move learning rate to state * add lr to opt state, some fixes in capture * fix optim * update tuple of containers as well * fix stream for compiled output * rng state for compile * nit * updates and comments --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com> 2024-02-07 17:29:22 -08:00			`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`