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

r2.14 cherry-pick: ea45e14 "Change jaxlib version to the next earliest version for MacOS + Linux CI builds."
ROCm · Nov 10, 2023 · 99d80a9 · 99d80a9
2 parents b4ee76a + aef24a9
commit 99d80a9
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'