Utilities API
=============

The utils module contains various utility functions and helpers.

Checkpointing
-------------

.. automodule:: utils.checkpoint
   :members:
   :undoc-members:
   :show-inheritance:

EMA (Exponential Moving Average)
--------------------------------

.. automodule:: utils.ema
   :members: EMA
   :show-inheritance:

Google Cloud Utilities
----------------------

.. automodule:: utils.gcloud_utils
   :members:
   :undoc-members:
   :show-inheritance:

Sharding
--------

.. automodule:: utils.sharding_utils
   :members:
   :undoc-members:
   :show-inheritance:

Visualization
-------------

.. automodule:: utils.visualize
   :members:
   :undoc-members:
   :show-inheritance: