optax.ScaleByRmsState
=====================

.. currentmodule:: optax

.. autoclass:: ScaleByRmsState