optax.amsgrad
=============

.. currentmodule:: optax

.. autofunction:: amsgrad