optax.scale\_by\_adam
=====================

.. currentmodule:: optax

.. autofunction:: scale_by_adam