optax.contrib.SophiaState
=========================

.. currentmodule:: optax.contrib

.. autoclass:: SophiaState