diff --git a/ml-notebook/cuda12-pytorch-2.2.2/requirements.txt b/ml-notebook/cuda12-pytorch-2.2.2/requirements.txt
index 32c6a3a..06462bb 100644
--- a/ml-notebook/cuda12-pytorch-2.2.2/requirements.txt
+++ b/ml-notebook/cuda12-pytorch-2.2.2/requirements.txt
@@ -29,3 +29,4 @@ timm
 transformers
 cmake
 lit
+jupyter-ai[all]