-
Notifications
You must be signed in to change notification settings - Fork 37
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Resolve default value of Handler.typestr if method missing.
#1457
opened Dec 24, 2024 by
copybara-service
bot
Loading…
Save
StepMetadata
in the AsyncCheckpointer
.
#1454
opened Dec 20, 2024 by
copybara-service
bot
Loading…
Fix adhoc reloads for the registery mechanism
#1453
opened Dec 20, 2024 by
copybara-service
bot
Loading…
Fix a bug where a custom snapshot_dir is not passed to _release_snapshot, causing snapshots to not be released in this case.
#1451
opened Dec 19, 2024 by
copybara-service
bot
Loading…
Add tests that don't require real accelerators to allow future reproducers.
#1445
opened Dec 16, 2024 by
copybara-service
bot
Loading…
Add logs in
consistent_restore_mesh
to examine the invariant.
#1427
opened Dec 5, 2024 by
copybara-service
bot
Loading…
Need to add because parent CL updates Orbax codebase.
#1369
opened Nov 21, 2024 by
copybara-service
bot
Loading…
Fix a bug in Orbax checkpointing where None values PyTree are not handled correctly. This is caused by a recent update in jax where None values are no longer considered as a leaf node: https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-34-october-4-2024.
#1286
opened Oct 31, 2024 by
copybara-service
bot
Loading…
Previous Next
ProTip!
What’s not been updated in a month: updated:<2024-11-24.