orbax
orbax copied to clipboard
Allow users to enable continuous checkpointing in Gemax Prod. Note that this is an experimental feature and should not be adopted by users without explicit approval.
Allow users to enable continuous checkpointing in Gemax Prod. Note that this is an experimental feature and should not be adopted by users without explicit approval.