Optimizers
==========

.. automodule:: numpyro.optim

Adam
----
.. autoclass:: numpyro.optim.Adam
   :members:
   :undoc-members:
   :inherited-members:

Adagrad
-------
.. autoclass:: numpyro.optim.Adagrad
   :members:
   :undoc-members:
   :inherited-members:

ClippedAdam
-----------
.. autoclass:: numpyro.optim.ClippedAdam
   :members:
   :undoc-members:
   :inherited-members:

Minimize
--------
.. autoclass:: numpyro.optim.Minimize
   :members:
   :undoc-members:
   :inherited-members:

Momentum
--------
.. autoclass:: numpyro.optim.Momentum
   :members:
   :undoc-members:
   :inherited-members:

RMSProp
-------
.. autoclass:: numpyro.optim.RMSProp
   :members:
   :undoc-members:
   :inherited-members:

RMSPropMomentum
---------------
.. autoclass:: numpyro.optim.RMSPropMomentum
   :members:
   :undoc-members:
   :inherited-members:

SGD
---
.. autoclass:: numpyro.optim.SGD
   :members:
   :undoc-members:
   :inherited-members:

SM3
---
.. autoclass:: numpyro.optim.SM3
   :members:
   :undoc-members:
   :inherited-members:

Optax support
-------------
.. autofunction:: numpyro.optim.optax_to_numpyro