From b8f1700fe910794bf01942db468a398dbee73ae3 Mon Sep 17 00:00:00 2001
From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com>
Date: Fri, 19 Jan 2024 20:54:33 +0000
Subject: [PATCH] Bump jupyterlab from 4.0.5 to 4.0.11

Bumps [jupyterlab](https://github.com/jupyterlab/jupyterlab) from 4.0.5 to 4.0.11.
- [Release notes](https://github.com/jupyterlab/jupyterlab/releases)
- [Changelog](https://github.com/jupyterlab/jupyterlab/blob/@jupyterlab/lsp@4.0.11/CHANGELOG.md)
- [Commits](https://github.com/jupyterlab/jupyterlab/compare/@jupyterlab/lsp@4.0.5...@jupyterlab/lsp@4.0.11)

---
updated-dependencies:
- dependency-name: jupyterlab
  dependency-type: indirect
...

Signed-off-by: dependabot[bot] <support@github.com>
---
 requirements-cuda.txt | 20 +++++++++++++-------
 requirements.txt      | 16 ++++++++++------
 2 files changed, 23 insertions(+), 13 deletions(-)

diff --git a/requirements-cuda.txt b/requirements-cuda.txt
index 9d3f25c5..20135fe6 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
@@ -223,6 +223,7 @@ jax[cuda,tpu] @ git+https://github.com/google/jax
     #   chex
     #   clu
     #   flax
+    #   jax
     #   optax
     #   orbax-checkpoint
     #   orbax-export
@@ -232,7 +233,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
@@ -304,7 +305,7 @@ jupyter-server==2.7.2
     #   notebook-shim
 jupyter-server-terminals==0.4.4
     # via jupyter-server
-jupyterlab==4.0.5
+jupyterlab==4.0.11
     # via notebook
 jupyterlab-pygments==0.2.2
     # via nbconvert
@@ -324,7 +325,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 +353,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 +442,7 @@ oauthlib==3.2.2
 opt-einsum==3.3.0
     # via
     #   jax
+    #   praxis
     #   tensorflow
 optax==0.1.7
     # via
@@ -508,7 +512,7 @@ prompt-toolkit==3.0.39
     # via
     #   ipython
     #   jupyter-console
-protobuf==3.20.3
+protobuf==3.19.6
     # via
     #   googleapis-common-protos
     #   grpcio-reflection
@@ -792,7 +796,9 @@ traitlets==5.9.0
     #   nbformat
     #   qtconsole
 transformers==4.32.0
-    # via t5
+    # via
+    #   -r requirements.in
+    #   t5
 typeguard==4.1.2
     # via
     #   jaxtyping
diff --git a/requirements.txt b/requirements.txt
index 7d077109..d91c2b88 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
@@ -221,6 +221,7 @@ jax[tpu] @ git+https://github.com/google/jax
     #   chex
     #   clu
     #   flax
+    #   jax
     #   optax
     #   orbax-checkpoint
     #   orbax-export
@@ -230,7 +231,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
@@ -302,7 +303,7 @@ jupyter-server==2.7.2
     #   notebook-shim
 jupyter-server-terminals==0.4.4
     # via jupyter-server
-jupyterlab==4.0.5
+jupyterlab==4.0.11
     # via notebook
 jupyterlab-pygments==0.2.2
     # via nbconvert
@@ -322,7 +323,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 +351,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 +440,7 @@ oauthlib==3.2.2
 opt-einsum==3.3.0
     # via
     #   jax
+    #   praxis
     #   tensorflow
 optax==0.1.7
     # via
@@ -506,7 +510,7 @@ prompt-toolkit==3.0.39
     # via
     #   ipython
     #   jupyter-console
-protobuf==3.20.3
+protobuf==3.19.6
     # via
     #   googleapis-common-protos
     #   grpcio-reflection