optax.scale\_by\_adamax
=======================

.. currentmodule:: optax

.. autofunction:: scale_by_adamax