xlstm_jax.models.xlstm_parallel.checkpointing
=============================================

.. py:module:: xlstm_jax.models.xlstm_parallel.checkpointing


Functions
---------

.. autoapisummary::

   xlstm_jax.models.xlstm_parallel.checkpointing.save_checkpoint


Module Contents
---------------

.. py:function:: save_checkpoint(state, log_dir)