diff --git a/Dockerfile b/Dockerfile
index 1df2ce160b96bae67fa60275685762daca6e0c2e..0ae54c4d9ee579afdfd283174d2b1f5641b96db2 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,7 +2,13 @@ FROM jupyter/tensorflow-notebook:hub-3.1.1
 
 USER root
 
-# Install Extensions
+# Update Pip and conda
 RUN pip3 install --upgrade pip
 RUN mamba update mamba
+
+# Conda packages
 RUN mamba install -c conda-forge control
+RUN mamba install -c conda-forge casadi
+
+# Install pip packages
+RUN pip install -c conda-forge jax
\ No newline at end of file