diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 1575c5ac86e7ed838bddca9115e60f2dbf396104..bf8aba8cbecbafc8b1b0ad4bded9091ba9178e0f 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -2,7 +2,7 @@ build-doc:
     image: jupyter/scipy-notebook
     stage: build
     script:
-        - pip install pdoc3
+        - pip install pdoc3 jax jaxlib
         - pdoc --html src
         - mkdir .public
         - mv html/src/* .public/