Merge pull request #62370 from tensorflow/r2.14-ea45e14c926

huberalberti · web-flow · commit 99d80a9e254c · 2023-11-10T13:47:47.000-08:00
r2.14 cherry-pick: ea45e14 "Change jaxlib version to the next earliest version for MacOS + Linux CI builds."
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'
