diff --git a/requirements-jax-cuda.txt b/requirements-jax-cuda.txt index b67a0f88b20..012313b75e0 100644 --- a/requirements-jax-cuda.txt +++ b/requirements-jax-cuda.txt @@ -4,7 +4,7 @@ tf2onnx # Torch cpu-only version (needed for testing). --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.9.0+cpu +torch==2.9.1+cpu # Jax with cuda support. --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html diff --git a/requirements-jax-tpu.txt b/requirements-jax-tpu.txt index 16c6a0a5900..193e04ee16f 100644 --- a/requirements-jax-tpu.txt +++ b/requirements-jax-tpu.txt @@ -4,7 +4,7 @@ tf2onnx # Torch cpu-only version (needed for testing). --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.6.0 +torch==2.9.1+cpu # Jax with cuda support. --find-links https://storage.googleapis.com/jax-releases/libtpu_releases.html diff --git a/requirements-tensorflow-cuda.txt b/requirements-tensorflow-cuda.txt index a5304051a3f..e96d425d83d 100644 --- a/requirements-tensorflow-cuda.txt +++ b/requirements-tensorflow-cuda.txt @@ -3,11 +3,11 @@ tensorflow[and-cuda]~=2.20.0 tf2onnx # This is the only version which works with TF 2.20.0. # TODO(#21914): Update this version when TF is updated. -ai-edge-litert==1.3.0 +ai-edge-litert==2.0.3 # Torch cpu-only version (needed for testing). --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.9.0+cpu +torch==2.9.1+cpu # Jax cpu-only version (needed for testing). jax[cpu] diff --git a/requirements-tensorflow-tpu.txt b/requirements-tensorflow-tpu.txt index aaac402056b..42d193d2d5b 100644 --- a/requirements-tensorflow-tpu.txt +++ b/requirements-tensorflow-tpu.txt @@ -1,11 +1,11 @@ --find-links https://storage.googleapis.com/libtpu-tf-releases/index.html -tensorflow-tpu==2.19.1 +tensorflow-tpu==2.20.0 tf2onnx # Torch cpu-only version (needed for testing). --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.6.0 +torch==2.9.1+cpu # Jax cpu-only version (needed for testing). jax diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index 455f5f00f05..a47f2469e7d 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -5,8 +5,8 @@ tf2onnx # Torch with cuda support. # - torch is pinned to a version that is compatible with torch-xla. --extra-index-url https://download.pytorch.org/whl/cu121 -torch==2.9.0 -torch-xla==2.8.1;sys_platform != 'darwin' +torch==2.9.1+cpu +torch-xla==2.9.0;sys_platform != 'darwin' # Jax cpu-only version (needed for testing). jax[cpu] diff --git a/requirements.txt b/requirements.txt index cacfc703da4..923a78671b3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,12 +6,12 @@ tensorflow~=2.20.0;sys_platform == 'darwin' tf2onnx # This is the only version which works with TF 2.20.0. # TODO(#21914): Update this version when TF is updated. -ai-edge-litert==1.3.0 +ai-edge-litert==2.0.3 # Torch. --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.6.0 -torch-xla==2.6.0;sys_platform != 'darwin' +torch==2.9.1+cpu +torch-xla==2.9.0;sys_platform != 'darwin' # Jax. # Pinned to 0.8.0 on CPU for CI compatibility with older backends.