optax.contrib.sam
=================

.. currentmodule:: optax.contrib

.. autofunction:: sam