diff --git a/pyproject.toml b/pyproject.toml
index b5eaa535..f616f400 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -9,6 +9,7 @@ keywords = ["explainable ai", "xai", "machine learning", "deep learning"]
 dependencies = [
     "numpy>=1.19.5",
     "torch>=1.13.1",
+    "captum>=0.6.0",
 ]
 dynamic = ["version"]