diff --git a/tensorflow/tools/ci_build/release/requirements_mac.txt b/tensorflow/tools/ci_build/release/requirements_mac.txt
index 0e0f832d088ea9..39349a3f3a6aa2 100644
--- 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
index f506e932516a04..8d7122076fcd91 100644
--- 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'