optax.ScaleBySM3State
=====================

.. currentmodule:: optax

.. autoclass:: ScaleBySM3State