optax.contrib.SAMState
======================

.. currentmodule:: optax.contrib

.. autoclass:: SAMState