[emergency checkpoint] Add `ReplicatorCheckpointManager` implementation for interoperating with replicator service provided by GKE (or theoretically, any other similar service).

google / orbax

Orbax provides common checkpointing and persistence utilities for JAX users

https://orbax.readthedocs.io/

Apache License 2.0

303 stars 36 forks source link

[emergency checkpoint] Add `ReplicatorCheckpointManager` implementation for interoperating with replicator service provided by GKE (or theoretically, any other similar service). #1339

Closed copybara-service[bot] closed 5 hours ago

copybara-service[bot] commented 1 week ago

[emergency checkpoint] Add ReplicatorCheckpointManager implementation for interoperating with replicator service provided by GKE (or theoretically, any other similar service).