optax.ema
=========

.. currentmodule:: optax

.. autofunction:: ema