orbax
orbax copied to clipboard
Add OCDBT support in Pax. If the feature is turned on, legacy checkpoints will be readable, but new checkpoints will be saved using OCDBT. A checkpoint version change is not required.
Add OCDBT support in Pax. If the feature is turned on, legacy checkpoints will be readable, but new checkpoints will be saved using OCDBT. A checkpoint version change is not required.