No public description

PiperOrigin-RevId: 700384094

.github/workflows/build.yml

@@ -80,7 +80,7 @@ jobs:
     strategy:
       matrix:
         python-version: ["3.10"]
-        jax-version: ["newest", "0.4.30"]  # keep in sync with minimum version in export/pyproject.toml
+        jax-version: ["newest","0.4.30", "0.4.34"]  # keep in sync with minimum version in export/pyproject.toml
     steps:
     - name: Cancel previous
       uses: styfle/[email protected]

export/CHANGELOG.md

@@ -10,7 +10,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 - Add `orbax_export.config_class` API to set global configs.
 - Add `orbax_export.typing` API
-- Require jax version >=0.4.30 as the jax.export API was introduced in this release.
+- Require jax version >=0.4.34 as the jax.export API was introduced in this release.
 - Add version flag to export_manager.py and a stubbed class to handle each version.
 - Adds a new module base class and two new module subclasses one for TensorFlow and
   one for Orbax.