orbax
orbax copied to clipboard
#v1 Add `LeafHandler` as a `CheckpointableHandler`, so that ordinary PyTree leaves can also be saved as individual checkpointables.
#v1 Add LeafHandler as a CheckpointableHandler, so that ordinary PyTree leaves can also be saved as individual checkpointables.