Change jaxlib version to the next earliest version for MacOS + Linux CI builds.

swachhandl · tensorflow-jenkins · commit aef24a9f5acb · 2023-11-10T21:45:40.000Z
PiperOrigin-RevId: 579282341
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'
