optax.scale\_by\_rms
====================

.. currentmodule:: optax

.. autofunction:: scale_by_rms