optax.ScaleState
================

.. currentmodule:: optax

.. autoclass:: ScaleState