Align requirments with Keras (keras-team#1386)

* Align requirments with Keras

* Update TF nightly versions

Author: sampathweb

requirements-jax-cuda.txt

@@ -1,6 +1,6 @@
 # Tensorflow cpu-only version.
-tf-nightly-cpu==2.16.0.dev20231109  # Pin a working nightly until rc0.
-tensorflow-text-nightly==2.16.0.dev20231109  # Pin a working nightly until rc0.
+tf-nightly-cpu==2.16.0.dev20231221  # Pin a working nightly until rc0.
+tensorflow-text-nightly==2.16.0.dev20231221  # Pin a working nightly until rc0.
 
 # Torch cpu-only version.
 --extra-index-url https://download.pytorch.org/whl/cpu

requirements-tensorflow-cuda.txt

@@ -1,7 +1,6 @@
 # Tensorflow with cuda support.
---extra-index-url https://pypi.nvidia.com
-tf-nightly[and-cuda]==2.16.0.dev20231109  # Pin a working nightly until rc0.
-tensorflow-text-nightly==2.16.0.dev20231109  # Pin a working nightly until rc0.
+tf-nightly[and-cuda]==2.16.0.dev20231221  # Pin a working nightly until rc0.
+tensorflow-text-nightly==2.16.0.dev20231221  # Pin a working nightly until rc0.
 
 # Torch cpu-only version.
 --extra-index-url https://download.pytorch.org/whl/cpu

requirements-torch-cuda.txt

@@ -1,11 +1,11 @@
 # Tensorflow cpu-only version.
-tf-nightly-cpu==2.16.0.dev20231109  # Pin a working nightly until rc0.
-tensorflow-text-nightly==2.16.0.dev20231109  # Pin a working nightly until rc0.
+tf-nightly-cpu==2.16.0.dev20231221  # Pin a working nightly until rc0.
+tensorflow-text-nightly==2.16.0.dev20231221  # Pin a working nightly until rc0.
 
 # Torch with cuda support.
---extra-index-url https://download.pytorch.org/whl/cu118
-torch==2.1.0
-torchvision==0.16.0
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.1.2
+torchvision==0.16.2
 
 # Jax cpu-only version.
 jax[cpu]

requirements.txt

@@ -1,6 +1,6 @@
 # Tensorflow.
-tf-nightly-cpu==2.16.0.dev20231109  # Pin a working nightly until rc0.
-tensorflow-text-nightly==2.16.0.dev20231109  # Pin a working nightly until rc0.
+tf-nightly-cpu==2.16.0.dev20231221  # Pin a working nightly until rc0.
+tensorflow-text-nightly==2.16.0.dev20231221  # Pin a working nightly until rc0.
 
 # Torch.
 --extra-index-url https://download.pytorch.org/whl/cpu