optax.ScaleByAmsgradState
=========================

.. currentmodule:: optax

.. autoclass:: ScaleByAmsgradState