orbax
orbax copied to clipboard
Fix slice selection logic in emergency checkpointing, so that a slice with a complete local checkpoint (if it exists) is always chosen as the secondary slice.
Fix slice selection logic in emergency checkpointing, so that a slice with a complete local checkpoint (if it exists) is always chosen as the secondary slice.