From 1c834139ef90cbebe98c2119aa02154cfe65e5cd Mon Sep 17 00:00:00 2001
From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com>
Date: Thu, 18 Jan 2024 16:25:35 +0000
Subject: [PATCH] Bump jupyter-lsp from 2.2.0 to 2.2.2

Bumps [jupyter-lsp](https://github.com/jupyter-lsp/jupyterlab-lsp) from 2.2.0 to 2.2.2.
- [Release notes](https://github.com/jupyter-lsp/jupyterlab-lsp/releases)
- [Changelog](https://github.com/jupyter-lsp/jupyterlab-lsp/blob/main/CHANGELOG.md)
- [Commits](https://github.com/jupyter-lsp/jupyterlab-lsp/commits)

---
updated-dependencies:
- dependency-name: jupyter-lsp
  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..0930ea1d 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
@@ -293,7 +294,7 @@ jupyter-events==0.7.0
     # via jupyter-server
 jupyter-http-over-ws==0.0.8
     # via lingvo
-jupyter-lsp==2.2.0
+jupyter-lsp==2.2.2
     # via jupyterlab
 jupyter-server==2.7.2
     # via
@@ -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..4dbebd26 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
@@ -291,7 +292,7 @@ jupyter-events==0.7.0
     # via jupyter-server
 jupyter-http-over-ws==0.0.8
     # via lingvo
-jupyter-lsp==2.2.0
+jupyter-lsp==2.2.2
     # via jupyterlab
 jupyter-server==2.7.2
     # via
@@ -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