diff --git a/requirements-doc.txt b/requirements-doc.txt
index d90d985e1..d88a0c02a 100644
--- a/requirements-doc.txt
+++ b/requirements-doc.txt
@@ -3,6 +3,7 @@ tqdm
 msgpack
 numba
 jax>=0.4.1
+jaxlib>=0.4.1
 matplotlib>=3.4
 scipy>=1.1.0
 numba