diff --git a/requirements-jax-cuda.txt b/requirements-jax-cuda.txt
index a2d69180b8..99157c6d66 100644
--- a/requirements-jax-cuda.txt
+++ b/requirements-jax-cuda.txt
@@ -1,5 +1,5 @@
 # Tensorflow cpu-only version.
-tf-nightly-cpu==2.16.0.dev20240104  # Pin a working nightly until rc0.
+tf-nightly-cpu==2.16.0.dev20240209  # Pin a working nightly until rc0.
 
 # Torch cpu-only version.
 --extra-index-url https://download.pytorch.org/whl/cpu
@@ -9,6 +9,6 @@ torchvision>=0.16.0
 # Jax with cuda support.
 # TODO: 0.4.24 has an updated Cuda version breaks Jax CI.
 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
-jax[cuda12_pip]==0.4.23
+jax[cuda12_pip]==0.4.24
 
 -r requirements-common.txt
\ No newline at end of file
diff --git a/requirements-tensorflow-cuda.txt b/requirements-tensorflow-cuda.txt
index 3889e83934..d27982d236 100644
--- a/requirements-tensorflow-cuda.txt
+++ b/requirements-tensorflow-cuda.txt
@@ -1,5 +1,5 @@
 # Tensorflow with cuda support.
-tf-nightly[and-cuda]==2.16.0.dev20240104  # Pin a working nightly until rc0.
+tf-nightly[and-cuda]==2.16.0.dev20240209  # Pin a working nightly until rc0.
 
 # Torch cpu-only version.
 --extra-index-url https://download.pytorch.org/whl/cpu
diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt
index 5bfbc37546..a86868f903 100644
--- a/requirements-torch-cuda.txt
+++ b/requirements-torch-cuda.txt
@@ -1,10 +1,10 @@
 # Tensorflow cpu-only version.
-tf-nightly-cpu==2.16.0.dev20240104  # Pin a working nightly until rc0.
+tf-nightly-cpu==2.16.0.dev20240209  # Pin a working nightly until rc0.
 
 # Torch with cuda support.
 --extra-index-url https://download.pytorch.org/whl/cu121
-torch==2.1.2+cu121
-torchvision==0.16.2+cu121
+torch==2.2.0+cu121
+torchvision==0.17.0+cu121
 
 # Jax cpu-only version.
 jax[cpu]
diff --git a/requirements.txt b/requirements.txt
index 02792b29b7..b4791d9109 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 # Tensorflow.
-tf-nightly-cpu==2.16.0.dev20240104  # Pin a working nightly until rc0.
+tf-nightly-cpu==2.16.0.dev20240209  # Pin a working nightly until rc0.
 
 # Torch.
 --extra-index-url https://download.pytorch.org/whl/cpu