optax.contrib.madgrad
=====================

.. currentmodule:: optax.contrib

.. autofunction:: madgrad