Skip to main content

Orbax Export

Project description

pip install orbax-export

import orbax.export

Orbax includes a serialization library for JAX users, enabling the exporting of JAX models to the TensorFlow SavedModel format.

To get started, check out our documentation.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

orbax-export-0.0.2.tar.gz (27.3 kB view hashes)

Uploaded Source

Built Distribution

orbax_export-0.0.2-py3-none-any.whl (45.6 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page