optax.adamax
============

.. currentmodule:: optax

.. autofunction:: adamax