orbax
orbax copied to clipboard
Introduce an enum `CheckpointingImpl` to simplify Pathways registration and to allow multiple options to be specified and resolved in order of priority.
Introduce an enum CheckpointingImpl to simplify Pathways registration and to allow multiple options to be specified and resolved in order of priority.