optax.contrib.MadgradState
==========================

.. currentmodule:: optax.contrib

.. autoclass:: MadgradState