flax\_extra.checkpoint package
==============================

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