.gitlab-ci: add jax

.gitlab-ci.yml

@@ -4,6 +4,7 @@ run:
   script:
     - pip install jupyter nbconvert
     - pip install matplotlib nifty8 pybind11
+    - pip install "jax[cpu]"
     - git clone --recursive https://gitlab.mpcdf.mpg.de/ift/resolve
     - cd resolve
     - pip install .