flax
Add `flax.nnx.remat` docstring
#4373
Merged

Loading