optax.update\_moment\_per\_elem\_norm
=====================================

.. currentmodule:: optax

.. autofunction:: update_moment_per_elem_norm