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 1f0b504a75a..002faf3ef64 100644 --- a/requirements-tensorflow-cuda.txt +++ b/requirements-tensorflow-cuda.txt @@ -5,7 +5,7 @@ ai-edge-litert # 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 b3046f1ee92..eb4e6c98257 100644 --- a/requirements.txt +++ b/requirements.txt @@ -8,8 +8,8 @@ ai-edge-litert # 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.