Update default JAX version requirement to 0.5.0.

PiperOrigin-RevId: 729313422

Author: BlaziusMaximus

checkpoint/CHANGELOG.md

@@ -7,6 +7,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.11.6] - 2025-02-20
+
 ### Added
 
 - Allow concurrent CheckpointManager.save requests and wait_until_finished.
@@ -19,6 +21,7 @@ background task. It also incorporates typing, in contrast with its predecessor.
 ## Changed
 
 - Move process_metadata to the checkpoint_num directory
+- Updated default JAX version requirement to 0.5.0.
 
 ## [0.11.5] - 2025-02-10
 

checkpoint/orbax/checkpoint/version.py

@@ -16,7 +16,7 @@
 
 # A new PyPI release will be pushed everytime `__version__` is increased.
 # Also modify version and date in CHANGELOG.
-__version__ = '0.11.5'
+__version__ = '0.11.6'
 
 
 # TODO: b/362813406 - Add latest change timestamp and commit number.

checkpoint/pyproject.toml

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