jax.custom\_remat
=================

.. currentmodule:: jax

.. autofunction:: custom_remat