jax.ad_checkpoint module

.. currentmodule:: jax.ad_checkpoint

.. automodule:: jax.ad_checkpoint

.. autosummary:: :toctree: _autosummary

checkpoint_name