diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index 4a8c340292a..b2fba4a8ea3 100644
--- a/.devcontainer/devcontainer.json
+++ b/.devcontainer/devcontainer.json
@@ -1,13 +1,21 @@
 {
-    "extensions": ["ms-python.python", "visualstudioexptteam.vscodeintellicode"],
-    "dockerFile": "Dockerfile",
-    "settings": {
-        "terminal.integrated.profiles.linux": {
-            "bash": {
-                "path": "/bin/bash"
+    "customizations":  {
+        "vscode": {
+            "extensions": [
+                "ms-python.python",
+                "ms-toolsai.jupyter",
+                "visualstudioexptteam.vscodeintellicode"
+            ],
+            "settings": {
+                "terminal.integrated.profiles.linux": {
+                    "bash": {
+                        "path": "/bin/bash"
+                    }
+                },
+                "terminal.integrated.defaultProfile.linux": "bash"
             }
-        },
-        "terminal.integrated.defaultProfile.linux": "bash"
+        }
     },
+    "dockerFile": "Dockerfile",
     "updateContentCommand": "pip install -e . pre-commit && pre-commit install"
 }