optax.update\_moment
====================

.. currentmodule:: optax

.. autofunction:: update_moment