Set JAX version requirement back to >= 0.5.0.

cpgaffney1 · Orbax Authors · commit 0b2ffde8ed62 · 2025-03-07T10:31:49.000-08:00
PiperOrigin-RevId: 734607391
diff --git a/checkpoint/CHANGELOG.md b/checkpoint/CHANGELOG.md
@@ -7,6 +7,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.11.8] - 2025-03-07
+
+### Changed
+
+- Set JAX version requirement back to >= 0.5.0.
+
 ## [0.11.7] - 2025-03-06
 
 ### Added
diff --git 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.7'
+__version__ = '0.11.8'
 # LINT.ThenChange(//depot//orbax/checkpoint/CHANGELOG.md)
 
 
diff --git 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.5.0',
     'numpy',
     'pyyaml',
     'tensorstore >= 0.1.71',
