Orbax Export
Project Links
Meta
Author: Orbax Authors
Requires Python: >=3.10
Classifiers
Development Status
- 4 - Beta
Intended Audience
- Developers
- Science/Research
License
- OSI Approved :: Apache Software License
Programming Language
- Python
Topic
- Scientific/Engineering :: Artificial Intelligence
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.
0.0.8
Sep 17, 2025
0.0.7
Jul 30, 2025
0.0.6
Nov 15, 2024
0.0.5
Jul 01, 2024
0.0.4
Jan 20, 2024
0.0.3
Aug 01, 2023
0.0.2
Apr 04, 2023
0.0.1
Feb 15, 2023
0.0.0
Feb 08, 2023
Wheel compatibility matrix
Files in release
Extras:
Dependencies:
absl-py
dataclasses-json
etils
jax
(>=0.4.34)
jaxlib
jaxtyping
numpy
protobuf
orbax-checkpoint
(>=0.9.0)