Add flax.nnx.remat docstring#4373
Merged
copybara-service[bot] merged 1 commit intogoogle:mainfrom 8bitmp3:add-nnx-remat-docstringNov 14, 2024
+15
Commits
Commits on Nov 14, 2024
- committed
flax.nnx.remat docstring#4373