Add CheckpointManager inheriting from Orbax that will eventually replace t5x Checkpointer. Currently, this class implements basic functionality and is tested by unit tests (gda_checkpoints_test), but is not actually used anywhere.
Add CheckpointManager inheriting from Orbax that will eventually replace t5x Checkpointer. Currently, this class implements basic functionality and is tested by unit tests (gda_checkpoints_test), but is not actually used anywhere.

We are missing core functionalities of checkpointing dataset iterators and supporting state_transformation_fns. Support for these features is upcoming.

In the future, we will continue to add more unit tests and integrate this CheckpointManager into T5X training code (flag-guarded).

PiperOrigin-RevId: 457791967

google-research/t5x