From aef24a9f5acb473aea177c3e07baf53c266bb553 Mon Sep 17 00:00:00 2001 From: Swachhand Lokhande Date: Fri, 3 Nov 2023 13:10:21 -0700 Subject: [PATCH] Change jaxlib version to the next earliest version for MacOS + Linux CI builds. PiperOrigin-RevId: 579282341 --- tensorflow/tools/ci_build/release/requirements_mac.txt | 2 +- tensorflow/tools/ci_build/release/requirements_ubuntu.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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'