2 Pull requests merged by 1 person

• Support Async Directory Creation in Composite Checkpoint Handler

  #1525 merged Jan 27, 2025

• Remove ThreadPoolExecutors from CheckpointHandlers and use futures.CommitFuture and futures.CommitFutureAwaitingContractedSignals that also support asynchronous directory creation.

  #1523 merged Jan 27, 2025

1 Pull request opened by 1 person

• Partially revert `replica_parallel` feature originally introduced by https://github.com/google/orbax/pull/1319, as performance regressions have been observed. Note also that simply setting `use_replica_parallel=False` does not fix the issue.

  #1541 opened Jan 27, 2025

1 Unresolved conversation

Sometimes conversations happen on old items that aren’t yet closed. Here is a list of all the Issues and Pull Requests with unresolved conversations.

• Fixes an issue with native serialization platform mismatches. It makes the following changes:

  #1538 commented on Jan 27, 2025 • 0 new comments