orbax
orbax copied to clipboard
Move implementation details of Orbax V1 `save_pytree` and `save_checkpointables` into helper file.
Move implementation details of Orbax V1 save_pytree and save_checkpointables into helper file.