optax.scale\_by\_amsgrad
========================

.. currentmodule:: optax

.. autofunction:: scale_by_amsgrad