jax.checkpoint\_policies.offload\_dot\_with\_no\_batch\_dims
============================================================

.. currentmodule:: jax

.. automethod:: checkpoint_policies.offload_dot_with_no_batch_dims