Improved the documentation

Author: Thomas-Rauter

README.md

@@ -51,10 +51,22 @@ Install `edge2torch` from PyPI with:
 pip install edge2torch
 ```
 
+To run the `interpret_model()` function:
+
+```bash
+pip install "edge2torch[interpret]"
+```
+
 For optional `AnnData` support:
 
 ```bash
-pip install "edge2torch[bio]"
+pip install "edge2torch[anndata]"
+```
+
+To install both optional extras:
+
+```bash
+pip install "edge2torch[all]"
 ```
 
 ## Minimal example

docs/feedforward-skip-edges.ipynb

@@ -47,10 +47,10 @@
    "id": "c76366f7",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-05-06T08:00:51.754135Z",
-     "iopub.status.busy": "2026-05-06T08:00:51.754028Z",
-     "iopub.status.idle": "2026-05-06T08:00:53.170874Z",
-     "shell.execute_reply": "2026-05-06T08:00:53.170199Z"
+     "iopub.execute_input": "2026-05-06T10:51:27.556150Z",
+     "iopub.status.busy": "2026-05-06T10:51:27.556042Z",
+     "iopub.status.idle": "2026-05-06T10:51:27.563780Z",
+     "shell.execute_reply": "2026-05-06T10:51:27.563298Z"
     }
    },
    "outputs": [],