diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index c49077bb27a2af158624e7222d999973487f0951..af6021a523ebb466bf17674c16f4251b70eb2578 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -131,6 +131,16 @@ pycodegen-integration:
     - py.test -v -n $NUM_CORES .
     - cd ../walberla/build/
     - make CodegenJacobiCPU CodegenJacobiGPU MicroBenchmarkGpuLbm LbCodeGenerationExample
+    - cd ..
+    - cd ../pystencils_autodiff
+    - apt install -y ninja-build
+    - python -m pip install -U cppimport pybind11
+    - python -m pip install --upgrade git+https://github.com/sympy/sympy.git
+    - python -m pip install --upgrade pip setuptools wheel
+    - python -m pip install --upgrade tensorflow==2.0
+    - python -m pip install -e .
+    - pytest
+    - python setup.py doctest
   tags:
     - docker
     - cuda