diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
index bf82c41..2162b4f 100644
--- a/.github/workflows/test.yaml
+++ b/.github/workflows/test.yaml
@@ -28,7 +28,7 @@ jobs:
       run: |
         sudo apt-get install valgrind
         python -m pip install --upgrade pip
-        python -m pip install jax
+        python -m pip install jax jaxlib
 
     - name: configure cmake
       run: cmake -B ${{github.workspace}}/build -DCMAKE_BUILD_TYPE=${{env.BUILD_TYPE}}