diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 9dd03ade9b022a08b0082865d34486998795983f..11a78ad2b542cb9b9c69e32b2e821bec42904903 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -2,7 +2,7 @@ Python 3 POCL:
   script:
   - export PY_EXE=python3
   - export PYOPENCL_TEST=portable:pthread
-  - export EXTRA_INSTALL="pyopencl mpi4py"
+  - export EXTRA_INSTALL="pyopencl mpi4py jax[cpu]"
   - curl -L -O https://gitlab.tiker.net/inducer/ci-support/raw/main/build-and-test-py-project.sh
   - ". ./build-and-test-py-project.sh"
   tags:
@@ -31,7 +31,7 @@ Python 3 POCL Examples:
 Pylint:
   script:
   - export PY_EXE=python3
-  - EXTRA_INSTALL="pyopencl mpi4py"
+  - EXTRA_INSTALL="pyopencl mpi4py jax[cpu] pygments"
   - curl -L -O https://gitlab.tiker.net/inducer/ci-support/raw/main/prepare-and-run-pylint.sh
   - . ./prepare-and-run-pylint.sh ${CI_PROJECT_NAME} test/test_*.py examples
   tags: