diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 60a82122f..1bf227287 100644 --- a/.github/workflows/build.yml +++ b/.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" diff --git a/checkpoint/CHANGELOG.md b/checkpoint/CHANGELOG.md index 593515975..658f3d2e9 100644 --- a/checkpoint/CHANGELOG.md +++ b/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 diff --git a/checkpoint/orbax/checkpoint/version.py b/checkpoint/orbax/checkpoint/version.py index a4a70c6d5..410afce39 100644 --- a/checkpoint/orbax/checkpoint/version.py +++ b/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) diff --git a/checkpoint/pyproject.toml b/checkpoint/pyproject.toml index 0ec1ffcd8..fc4dc166b 100644 --- a/checkpoint/pyproject.toml +++ b/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',