orbax
orbax copied to clipboard
Add Checkpointer/CheckpointManager implementations that rely on t5x.checkpoints.Checkpointer to execute save and restore. This refactor should not change behavior.
Add Checkpointer/CheckpointManager implementations that rely on t5x.checkpoints.Checkpointer to execute save and restore. This refactor should not change behavior.
This represents an intermediate stage before creation of more generic APIs not specific to T5X.