Orbax Export
Project description
Orbax Export
pip install orbax-export (latest PyPi release) OR
pip install 'git+https://github.com/google/orbax/#subdirectory=export' (from this repository, at HEAD)
import orbax.export
Orbax includes a serialization library for JAX users, enabling the exporting of JAX models to the TensorFlow SavedModel format.
Note that orbax-export requires TensorFlow, but does not include it by default to allow for flexibility in version choice. If you wish to install with standard TensorFlow, please use pip install orbax-export[all].
To get started, check out our documentation.
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.3.tar.gz
(33.7 kB
view hashes)
Built Distribution
Close
Hashes for orbax_export-0.0.3-py3-none-any.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | a0e391a01c9935c28cf03fa23ecb5009ca64ad45f15dadb8a21b790886167048 |
|
| MD5 | 41c16d3489e30a02d0a056e4d704d09a |
|
| BLAKE2b-256 | 360360feed9837fd57fb3067a492e67f03b07252b4a3208d2dfd14f800d920be |
