diff --git a/docs/requirements.txt b/docs/requirements.txt
index 536f10ed6..b7e605bcd 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -1,19 +1,16 @@
-# sphinx <3 is required by myst-parser[sphinx] v0.9.1, the latest version as of 2020-08-05
-sphinx >=2.1, <3
-
+# sphinx <4 required by myst-nb v0.12.0 (Feb 2021)
+# sphinx >=3 required by sphinx-autodoc-typehints v1.11.1 (Oct 2020)
+sphinx >=3, <4
 sphinx_rtd_theme
-
-# sphinx-autodoc-typehints 1.11 requires sphinx >= 3, but we can't have that right now.
-sphinx-autodoc-typehints <1.11
-
-jaxlib
-ipykernel
+sphinx-autodoc-typehints
 myst-nb
-# The next packages are for notebooks
-matplotlib
-sklearn
-# For CI tests.
+
+# Packages used for CI tests.
 pytest
 pytest-xdist
-# Must install jax itself for notebook execution to work
-.
+
+# Packages used for notebook execution
+matplotlib
+sklearn
+jaxlib
+.  # install jax itself from current directory.