Update version numbers from requirements.txt

This eradicates the protobuf version conflict.

PiperOrigin-RevId: 671484894

requirements/requirements.txt

@@ -1,14 +1,14 @@
-absl-py>=0.13.0
-attrs>=21.4.0
-chex>=0.0.8
-dm-haiku>=0.0.4
-jax>=0.2.18
-jaxlib>=0.1.69
+absl-py>=2.1.0
+attrs>=24.2.0
+chex>=0.1.86
+dm-haiku>=0.0.12
+jax>=0.4.31
+jaxlib>=0.4.31
 ml_collections>=0.1.1
-numpy>=1.21.1
+numpy>=1.26.4
 opt-einsum>=3.3.0
-optax>=0.0.9
+optax>=0.2.3
 six>=1.16.0
-tensorflow>=2.9.0
-tfds-nightly==4.5.2.dev202204190046
-toolz>=0.11.1
+tensorflow>=2.17.0
+tfds-nightly>=4.9.6.dev202409060044
+toolz>=0.12.1