Commit 0804444

committed

Update checkpointing guide

1 parent b0e18a5 commit 0804444Copy full SHA for 0804444

4 files changed

+136

-71

lines changed

docs_nnx/guides
- checkpointing.ipynb
- checkpointing.md
flax
- jax_utils.py
tests/nnx
- mutable_array_test.py

4 files changed

+136

-71

lines changed

Comments

(0)