diff --git a/Dockerfile b/Dockerfile
index edc262ab9c808bc69ea232b860786032a2e3535a..f3688cdc60e0e96eb2ea0e41a0d89db7b0d6553a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -16,6 +16,7 @@ RUN mamba install -c conda-forge control
 RUN mamba install -c conda-forge casadi
 RUN mamba install -c conda-forge networkx
 RUN mamba install -c conda-forge pillow
+RUN mamba install -c conda-forge jaxlib
 RUN mamba install -c conda-forge jax
 RUN mamba install -c conda-forge flax