Add save barrier to ensure checkpoint manager and checkpointers save start is synchronized on all processes.#1629

Open

copybara-service[bot] wants to merge 1 commit intomainfrom test_730721643

+54-2

Commits on Mar 6, 2025

Add save barrier to ensure checkpoint manager and checkpointers save start is synchronized on all processes.

mridul-sahu
authored and
Orbax Authors
committed