Change jaxlib version to the next earliest version for MacOS + Linux …

…CI builds. PiperOrigin-RevId: 579282341
tensorflow · Nov 10, 2023 · aef24a9 · aef24a9
1 parent b4ee76a
commit aef24a9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tensorflow/tools/ci_build/release/requirements_mac.txt b/tensorflow/tools/ci_build/release/requirements_mac.txt
@@ -9,4 +9,4 @@ setuptools
 
 # Test dependencies which don't exist on Windows
 jax ~= 0.3.24
-jaxlib ~= 0.3.24 # Earliest version for Python 3.11
+jaxlib ~= 0.4.1
diff --git a/tensorflow/tools/ci_build/release/requirements_ubuntu.txt b/tensorflow/tools/ci_build/release/requirements_ubuntu.txt
@@ -6,4 +6,4 @@ PyYAML ~= 6.0
 
 # Test dependencies which don't exist on Windows
 jax ~= 0.3.14
-jaxlib ~= 0.3.14; platform.machine != 'aarch64'
+jaxlib ~= 0.4.1; platform.machine != 'aarch64'