Bump the python group with 5 updates (#2340)

Bumps the python group with 5 updates:

| Package | From | To |
| --- | --- | --- |
| [tf-nightly-cpu](https://github.com/tensorflow/tensorflow) | `2.16.0.dev20240104` | `2.16.0.dev20240209` |
| torch | `2.1.2+cu121` | `2.2.0+cu121` |
| torchvision | `0.16.2+cu121` | `0.17.0+cu121` |
| [tf-nightly[and-cuda]](https://github.com/tensorflow/tensorflow) | `2.16.0.dev20240104` | `2.16.0.dev20240209` |
| [jax[cuda12_pip]](https://github.com/google/jax) | `0.4.23` | `0.4.24` |


Updates `tf-nightly-cpu` from 2.16.0.dev20240104 to 2.16.0.dev20240209
- [Release notes](https://github.com/tensorflow/tensorflow/releases)
- [Changelog](https://github.com/tensorflow/tensorflow/blob/master/RELEASE.md)
- [Commits](https://github.com/tensorflow/tensorflow/commits)

Updates `torch` from 2.1.2+cu121 to 2.2.0+cu121

Updates `torchvision` from 0.16.2+cu121 to 0.17.0+cu121

Updates `tf-nightly[and-cuda]` from 2.16.0.dev20240104 to 2.16.0.dev20240209
- [Release notes](https://github.com/tensorflow/tensorflow/releases)
- [Changelog](https://github.com/tensorflow/tensorflow/blob/master/RELEASE.md)
- [Commits](https://github.com/tensorflow/tensorflow/commits)

Updates `jax[cuda12_pip]` from 0.4.23 to 0.4.24
- [Release notes](https://github.com/google/jax/releases)
- [Changelog](https://github.com/google/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@jax-v0.4.23...jax-v0.4.24)

---
updated-dependencies:
- dependency-name: tf-nightly-cpu
  dependency-type: direct:production
  update-type: version-update:semver-patch
  dependency-group: python
- dependency-name: torch
  dependency-type: direct:production
  update-type: version-update:semver-minor
  dependency-group: python
- dependency-name: torchvision
  dependency-type: direct:production
  update-type: version-update:semver-minor
  dependency-group: python
- dependency-name: tf-nightly[and-cuda]
  dependency-type: direct:production
  update-type: version-update:semver-patch
  dependency-group: python
- dependency-name: jax[cuda12_pip]
  dependency-type: direct:production
  update-type: version-update:semver-patch
  dependency-group: python
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

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

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

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]

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