diff --git a/requirements-cuda.txt b/requirements-cuda.txt
index 9d3f25c5..e51d3a9b 100644
--- a/requirements-cuda.txt
+++ b/requirements-cuda.txt
@@ -140,7 +140,7 @@ filelock==3.12.2
     #   transformers
 flatbuffers==1.12
     # via tensorflow
-flax==0.7.2
+flax @ git+https://github.com/google/flax
     # via
     #   clu
     #   paxml
@@ -185,8 +185,10 @@ grpcio-reflection==1.48.2
     # via -r requirements.in
 h5py==3.9.0
     # via tensorflow
-huggingface-hub==0.16.4
-    # via transformers
+huggingface-hub==0.20.1
+    # via
+    #   tokenizers
+    #   transformers
 idna==3.4
     # via
     #   anyio
@@ -232,7 +234,7 @@ jax[cuda,tpu] @ git+https://github.com/google/jax
     #   seqio-nightly
 jax-bitempered-loss==0.0.2
     # via praxis
-jaxlib==0.4.14+cuda11.cudnn86
+jaxlib==0.4.23+cuda11.cudnn86
     # via
     #   chex
     #   clu
@@ -324,7 +326,7 @@ libclang==16.0.6
     # via tensorflow
 libcst==1.0.1
     # via fiddle
-libtpu-nightly==0.1.dev20230727
+libtpu-nightly==0.1.dev20231213
     # via jax
 lingvo==0.12.7
     # via
@@ -352,7 +354,9 @@ matplotlib-inline==0.1.6
 mdurl==0.1.2
     # via markdown-it-py
 mesh-tensorflow[transformer]==0.1.21
-    # via t5
+    # via
+    #   mesh-tensorflow
+    #   t5
 mistune==3.0.1
     # via nbconvert
 ml-collections==0.1.1
@@ -439,6 +443,7 @@ oauthlib==3.2.2
 opt-einsum==3.3.0
     # via
     #   jax
+    #   praxis
     #   tensorflow
 optax==0.1.7
     # via
@@ -508,7 +513,7 @@ prompt-toolkit==3.0.39
     # via
     #   ipython
     #   jupyter-console
-protobuf==3.20.3
+protobuf==3.19.6
     # via
     #   googleapis-common-protos
     #   grpcio-reflection
@@ -748,7 +753,7 @@ threadpoolctl==3.2.0
     # via scikit-learn
 tinycss2==1.2.1
     # via nbconvert
-tokenizers==0.13.3
+tokenizers==0.15.0
     # via transformers
 toml==0.10.2
     # via
@@ -791,8 +796,10 @@ traitlets==5.9.0
     #   nbconvert
     #   nbformat
     #   qtconsole
-transformers==4.32.0
-    # via t5
+transformers==4.36.0
+    # via
+    #   -r requirements.in
+    #   t5
 typeguard==4.1.2
     # via
     #   jaxtyping
diff --git a/requirements.txt b/requirements.txt
index 7d077109..fa8c452b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -139,7 +139,7 @@ filelock==3.12.2
     #   transformers
 flatbuffers==1.12
     # via tensorflow
-flax==0.7.2
+flax @ git+https://github.com/google/flax
     # via
     #   clu
     #   paxml
@@ -184,8 +184,10 @@ grpcio-reflection==1.48.2
     # via -r requirements.in
 h5py==3.9.0
     # via tensorflow
-huggingface-hub==0.16.4
-    # via transformers
+huggingface-hub==0.20.1
+    # via
+    #   tokenizers
+    #   transformers
 idna==3.4
     # via
     #   anyio
@@ -230,7 +232,7 @@ jax[tpu] @ git+https://github.com/google/jax
     #   seqio-nightly
 jax-bitempered-loss==0.0.2
     # via praxis
-jaxlib==0.4.14
+jaxlib==0.4.23
     # via
     #   chex
     #   clu
@@ -322,7 +324,7 @@ libclang==16.0.6
     # via tensorflow
 libcst==1.0.1
     # via fiddle
-libtpu-nightly==0.1.dev20230727
+libtpu-nightly==0.1.dev20231213
     # via jax
 lingvo==0.12.7
     # via
@@ -350,7 +352,9 @@ matplotlib-inline==0.1.6
 mdurl==0.1.2
     # via markdown-it-py
 mesh-tensorflow[transformer]==0.1.21
-    # via t5
+    # via
+    #   mesh-tensorflow
+    #   t5
 mistune==3.0.1
     # via nbconvert
 ml-collections==0.1.1
@@ -437,6 +441,7 @@ oauthlib==3.2.2
 opt-einsum==3.3.0
     # via
     #   jax
+    #   praxis
     #   tensorflow
 optax==0.1.7
     # via
@@ -506,7 +511,7 @@ prompt-toolkit==3.0.39
     # via
     #   ipython
     #   jupyter-console
-protobuf==3.20.3
+protobuf==3.19.6
     # via
     #   googleapis-common-protos
     #   grpcio-reflection
@@ -746,7 +751,7 @@ threadpoolctl==3.2.0
     # via scikit-learn
 tinycss2==1.2.1
     # via nbconvert
-tokenizers==0.13.3
+tokenizers==0.15.0
     # via transformers
 toml==0.10.2
     # via
@@ -789,7 +794,7 @@ traitlets==5.9.0
     #   nbconvert
     #   nbformat
     #   qtconsole
-transformers==4.32.0
+transformers==4.36.0
     # via
     #   -r requirements.in
     #   t5