PR #3002 Update checkpointing guide with Orbax APIs

Update checkpointing guide with Orbax APIs #3002

copybara-service merged 1 commit into google:main from IvyZX:orbax-upgrade-guide

IvyZX requested a review from

cgarciae 2 years ago

IvyZX requested a review from

8bitmp3 2 years ago

IvyZX assigned

IvyZX 2 years ago

8bitmp3 approved these changes on 2023-04-12

Update Flax/Orbax save and load checkpoints

6b07e09f

IvyZX force pushed to 6b07e09f 2 years ago

IvyZX added pull ready

copybara-service merged 8f55865b into main 2 years ago

cgarciae approved these changes on 2023-04-14

IvyZX deleted the orbax-upgrade-guide branch 312 days ago

Reviewers

cgarciae

8bitmp3

Assignees

IvyZX

Labels

pull ready

Milestone

No milestone