diff --git a/pyproject.toml b/pyproject.toml
index 3c8bb401..736bf9f5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "lit-nlp"
-version = "1.0.1"
+version = "1.0.2"
 authors = [
   { name="Google, LLC", email="lit-dev@google.com" }
 ]
diff --git a/requirements_examples.txt b/requirements_examples.txt
index 87d542b2..e078388b 100644
--- a/requirements_examples.txt
+++ b/requirements_examples.txt
@@ -18,7 +18,7 @@ sentencepiece==0.1.99
 tensorflow-datasets==4.8.0
 torch==2.0.1
 transformers==4.27.1
-jax==0.3.16
-jaxlib==0.3.15
+jax==0.4.1
+jaxlib==0.4.1
 flax==0.5.3
 # LINT.ThenChange(./pyproject.toml)