mattertune.callbacks.ema
========================

.. automodule:: mattertune.callbacks.ema

   
   .. rubric:: Functions

   .. autosummary::
   
      ema_update
      run_ema_update_cpu
   
   .. rubric:: Classes

   .. autosummary::
   
      EMA
      EMAConfig
      EMAOptimizer