Orbax
Project description
Orbax
Orbax is no longer a standalone package, and should not be installed directly.
Orbax is a namespace providing common utility libraries for JAX users.
Please install orbax-checkpoint or
orbax-export instead, for model
checkpointing and exporting utilities respectively. By default, installing
orbax
will install orbax-checkpoint
.
Support
Contact orbax-dev@google.com for help or with any questions about Orbax!
History
Orbax was initially published as a catch-all package itself. In order to
minimize dependency bloat for users, we have frozen that package at
orbax-0.1.9
, and will continue to release future changes under the
domain-specific utilities listed above (e.g. orbax-checkpoint
).
As we have preserved the orbax namespace, existing import statements can remain
unchanged (e.g. from orbax import checkpoint
).
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.