Bump minimum JAX version to v0.6.0

.github/workflows/build.yml

@@ -34,7 +34,7 @@ jobs:
         jax-version: ["newest"]
         include:
           - python-version: "3.10"
-            jax-version: "0.5.0"  # keep in sync with minimum version in checkpoint/pyproject.toml
+            jax-version: "0.6.0"  # keep in sync with minimum version in checkpoint/pyproject.toml
           # TODO(b/401258175) Re-enable once JAX nightlies are fixed.
           # - python-version: "3.13"
           #   jax-version: "nightly"

checkpoint/CHANGELOG.md

@@ -7,6 +7,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.11.25] - 2025-09-10
+
 ### Changed
 
 - Move temporary path detection into `TemporaryPath` class. This class serves
@@ -17,6 +19,7 @@ paths.
 - Remove the `jax_pmap_no_rank_reduction=False` code path for
 `utils.fully_replicated_host_local_array_to_global_array`.
 `jax_pmap_no_rank_reduction=True` by default in JAX.
+- Changed to minimum jax version requirement to v0.6.0
 
 ## [0.11.24] - 2025-08-28
 

checkpoint/orbax/checkpoint/version.py

@@ -17,7 +17,7 @@
 # A new PyPI release will be pushed everytime `__version__` is increased.
 # Also modify version and date in CHANGELOG.
 # LINT.IfChange
-__version__ = '0.11.24'
+__version__ = '0.11.25'
 # LINT.ThenChange(//depot//orbax/checkpoint/CHANGELOG.md)
 
 

checkpoint/pyproject.toml

@@ -26,7 +26,7 @@ dependencies = [
     'etils[epath,epy]',
     'typing_extensions',
     'msgpack',
-    'jax >= 0.5.0',
+    'jax >= 0.6.0',
     'numpy',
     'pyyaml',
     'tensorstore >= 0.1.71',