Merge pull request #3160 from levskaya:fix_doc_reqs

PiperOrigin-RevId: 542702348
google · Jun 22, 2023 · 50684ad · 50684ad
2 parents 11277dc + abfd7e7
commit 50684ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,7 +1,7 @@
 sphinx>=3.3.1
 sphinx-book-theme
 Pygments>=2.6.1
-jax>=0.3
+jax>=0.4
 jaxlib
 ipykernel
 myst_nb