-
Notifications
You must be signed in to change notification settings - Fork 39
Pull requests: google/orbax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Remove
_CommitFuture
from type_handlers.py and use futures.CommitFutureAwaitingContractedSignals
that also support asynchronous directory creation.
#1522
opened Jan 22, 2025 by
copybara-service
bot
Loading…
Move
NoopFuture
, ChainedFuture
and ThreadRaisingException
from orbax.checkpoint.futures
to orbax.checkpoint._src.futures
#1521
opened Jan 22, 2025 by
copybara-service
bot
Loading…
Add new Future that waits for all awaitable directory signals to be set before proceeding with the commit.
#1518
opened Jan 21, 2025 by
copybara-service
bot
Loading…
Add
SaveIntervalPolicy
to better encapsulate various options around choosing whether or not to perform a save at a particular step. Added a policy that allows for checkpointing as often as possible, as long as a save is not already in progress (continuous checkpointing).
#1515
opened Jan 17, 2025 by
copybara-service
bot
Loading…
Add support for merging array groups across multiple devices, with uneven sharding between devices.
#1514
opened Jan 17, 2025 by
copybara-service
bot
Loading…
Remove items_and_handlers-related
CompositeCheckpointHandler
code.
#1512
opened Jan 16, 2025 by
copybara-service
bot
Loading…
Use
RootMetadata
and StepMetadata
in CheckpointManager
.
#1511
opened Jan 16, 2025 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…
Previous Next
ProTip!
Add no:assignee to see everything that’s not assigned.