flax.linen.remat\_scan ====================== .. currentmodule:: flax.linen .. autofunction:: remat_scan