optax.contrib.scale\_by\_madgrad
================================

.. currentmodule:: optax.contrib

.. autofunction:: scale_by_madgrad