diff --git a/nbs/adaptive_discriminator_augmentation_preset.ipynb b/nbs/adaptive_discriminator_augmentation_preset.ipynb
index a288f15..aef5ab0 100644
--- a/nbs/adaptive_discriminator_augmentation_preset.ipynb
+++ b/nbs/adaptive_discriminator_augmentation_preset.ipynb
@@ -42,6 +42,12 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:24.041970Z",
+     "iopub.status.busy": "2026-03-20T00:35:24.041834Z",
+     "iopub.status.idle": "2026-03-20T00:35:25.415061Z",
+     "shell.execute_reply": "2026-03-20T00:35:25.414513Z"
+    },
     "id": "AM7V1IZiQzfG"
    },
    "outputs": [],
@@ -63,6 +69,12 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:25.417753Z",
+     "iopub.status.busy": "2026-03-20T00:35:25.417425Z",
+     "iopub.status.idle": "2026-03-20T00:35:26.242382Z",
+     "shell.execute_reply": "2026-03-20T00:35:26.242029Z"
+    },
     "id": "X16VOLFWQJs0"
    },
    "outputs": [],
@@ -111,6 +123,12 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:26.244222Z",
+     "iopub.status.busy": "2026-03-20T00:35:26.244087Z",
+     "iopub.status.idle": "2026-03-20T00:35:27.342674Z",
+     "shell.execute_reply": "2026-03-20T00:35:27.342276Z"
+    },
     "id": "ZHOopcuY7Lbr"
    },
    "outputs": [],
@@ -136,10 +154,31 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:27.344703Z",
+     "iopub.status.busy": "2026-03-20T00:35:27.344489Z",
+     "iopub.status.idle": "2026-03-20T00:35:27.365959Z",
+     "shell.execute_reply": "2026-03-20T00:35:27.365555Z"
+    },
     "id": "CRp0FV_z7Shh"
    },
    "outputs": [],
-   "source": "panda_image = kornia.image_to_tensor(np.array(Image.open(\"panda.jpg\").convert(\"RGB\"))).float() / 255.0  # C, H, W\npanda_images = panda_image.unsqueeze(0).repeat(8, 1, 1, 1)  # 8, C, H, W\n\n\ndef plot_images(images_tensor, n_rows=4, figscale=1):\n    \"\"\"utility function to plot expects a B, C, H, W tensor and plot them as a grid\"\"\"\n    b, c, h, w = images_tensor.shape\n    assert len(images_tensor) % n_rows == 0\n    n_cols = len(images_tensor) // n_rows\n    images_tensor = images_tensor.view(n_rows, n_cols, c, h, w).permute(2, 0, 3, 1, 4).reshape(c, h * n_rows, w * n_cols)\n    plt.figure(figsize=(h // figscale, w // figscale))\n    plt.imshow(kornia.tensor_to_image(images_tensor))\n    plt.axis(\"off\")\n    plt.show()"
+   "source": [
+    "panda_image = kornia.image_to_tensor(np.array(Image.open(\"panda.jpg\").convert(\"RGB\"))).float() / 255.0  # C, H, W\n",
+    "panda_images = panda_image.unsqueeze(0).repeat(8, 1, 1, 1)  # 8, C, H, W\n",
+    "\n",
+    "\n",
+    "def plot_images(images_tensor, n_rows=4, figscale=1):\n",
+    "    \"\"\"utility function to plot expects a B, C, H, W tensor and plot them as a grid\"\"\"\n",
+    "    b, c, h, w = images_tensor.shape\n",
+    "    assert len(images_tensor) % n_rows == 0\n",
+    "    n_cols = len(images_tensor) // n_rows\n",
+    "    images_tensor = images_tensor.view(n_rows, n_cols, c, h, w).permute(2, 0, 3, 1, 4).reshape(c, h * n_rows, w * n_cols)\n",
+    "    plt.figure(figsize=(h // figscale, w // figscale))\n",
+    "    plt.imshow(kornia.tensor_to_image(images_tensor))\n",
+    "    plt.axis(\"off\")\n",
+    "    plt.show()"
+   ]
   },
   {
    "cell_type": "code",
@@ -149,6 +188,12 @@
      "base_uri": "https://localhost:8080/",
      "height": 200
     },
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:27.367861Z",
+     "iopub.status.busy": "2026-03-20T00:35:27.367731Z",
+     "iopub.status.idle": "2026-03-20T00:35:27.686390Z",
+     "shell.execute_reply": "2026-03-20T00:35:27.685767Z"
+    },
     "id": "bfIU-mQFtUzW",
     "outputId": "a65e6ed3-9aef-47b2-bd0a-bd756d656de3"
    },
@@ -179,6 +224,12 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:27.717817Z",
+     "iopub.status.busy": "2026-03-20T00:35:27.717639Z",
+     "iopub.status.idle": "2026-03-20T00:35:27.722181Z",
+     "shell.execute_reply": "2026-03-20T00:35:27.721876Z"
+    },
     "id": "js_sWEDvRjWL"
    },
    "outputs": [],
@@ -194,6 +245,12 @@
      "base_uri": "https://localhost:8080/",
      "height": 398
     },
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:27.723676Z",
+     "iopub.status.busy": "2026-03-20T00:35:27.723585Z",
+     "iopub.status.idle": "2026-03-20T00:35:28.358888Z",
+     "shell.execute_reply": "2026-03-20T00:35:28.358483Z"
+    },
     "id": "ylqtkhXMTAH0",
     "outputId": "db880256-9d30-4fc5-92b7-801436789c39"
    },
@@ -208,6 +265,12 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:28.369465Z",
+     "iopub.status.busy": "2026-03-20T00:35:28.369247Z",
+     "iopub.status.idle": "2026-03-20T00:35:28.372530Z",
+     "shell.execute_reply": "2026-03-20T00:35:28.372260Z"
+    },
     "id": "3WdFmmi6fqKA"
    },
    "outputs": [],
@@ -229,6 +292,12 @@
      "base_uri": "https://localhost:8080/",
      "height": 381
     },
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:28.373857Z",
+     "iopub.status.busy": "2026-03-20T00:35:28.373775Z",
+     "iopub.status.idle": "2026-03-20T00:35:28.739155Z",
+     "shell.execute_reply": "2026-03-20T00:35:28.738851Z"
+    },
     "id": "_x8uvJnzvaax",
     "outputId": "4e0805b3-1872-4930-b00d-5ebdb772cb91"
    },
@@ -261,6 +330,12 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:28.747056Z",
+     "iopub.status.busy": "2026-03-20T00:35:28.746872Z",
+     "iopub.status.idle": "2026-03-20T00:35:28.753805Z",
+     "shell.execute_reply": "2026-03-20T00:35:28.753507Z"
+    },
     "id": "Q_uzg2nLge7l"
    },
    "outputs": [],
@@ -300,6 +375,12 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:28.755104Z",
+     "iopub.status.busy": "2026-03-20T00:35:28.755020Z",
+     "iopub.status.idle": "2026-03-20T00:35:28.758024Z",
+     "shell.execute_reply": "2026-03-20T00:35:28.757790Z"
+    },
     "id": "gu906vdhjtfl"
    },
    "outputs": [],
@@ -339,10 +420,25 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:28.759353Z",
+     "iopub.status.busy": "2026-03-20T00:35:28.759271Z",
+     "iopub.status.idle": "2026-03-20T00:35:28.761440Z",
+     "shell.execute_reply": "2026-03-20T00:35:28.761164Z"
+    },
     "id": "tq1g9FEAyAjH"
    },
    "outputs": [],
-   "source": "def create_batch(images_folder):\n    images = []\n    for image_file in os.listdir(images_folder):\n        image = kornia.image_to_tensor(np.array(Image.open(os.path.join(images_folder, image_file)).convert(\"RGB\")))\n        image = (image.float() / 127.5) - 1  # normalizing to be in (-1, 1) as the generator output\n        images.append(image)\n    batch = torch.stack(images, dim=0)\n    return batch"
+   "source": [
+    "def create_batch(images_folder):\n",
+    "    images = []\n",
+    "    for image_file in os.listdir(images_folder):\n",
+    "        image = kornia.image_to_tensor(np.array(Image.open(os.path.join(images_folder, image_file)).convert(\"RGB\")))\n",
+    "        image = (image.float() / 127.5) - 1  # normalizing to be in (-1, 1) as the generator output\n",
+    "        images.append(image)\n",
+    "    batch = torch.stack(images, dim=0)\n",
+    "    return batch"
+   ]
   },
   {
    "cell_type": "code",
@@ -351,6 +447,12 @@
     "colab": {
      "base_uri": "https://localhost:8080/"
     },
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:28.762943Z",
+     "iopub.status.busy": "2026-03-20T00:35:28.762835Z",
+     "iopub.status.idle": "2026-03-20T00:35:29.502345Z",
+     "shell.execute_reply": "2026-03-20T00:35:29.502036Z"
+    },
     "id": "8dGez6UhoUuK",
     "outputId": "080b5600-5ccc-4393-99d0-45b34ecfdc79"
    },
@@ -397,6 +499,12 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:29.503887Z",
+     "iopub.status.busy": "2026-03-20T00:35:29.503712Z",
+     "iopub.status.idle": "2026-03-20T00:35:29.507204Z",
+     "shell.execute_reply": "2026-03-20T00:35:29.506966Z"
+    },
     "id": "4ViQbI5LCdBS"
    },
    "outputs": [],
@@ -426,6 +534,12 @@
     "colab": {
      "base_uri": "https://localhost:8080/"
     },
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:35:29.508751Z",
+     "iopub.status.busy": "2026-03-20T00:35:29.508598Z",
+     "iopub.status.idle": "2026-03-20T00:37:45.959164Z",
+     "shell.execute_reply": "2026-03-20T00:37:45.958698Z"
+    },
     "id": "Luo-sw4yppKE",
     "outputId": "da5e9412-8870-40ff-f9e3-4577a345da33"
    },
@@ -501,6 +615,12 @@
      "base_uri": "https://localhost:8080/",
      "height": 693
     },
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:37:45.961451Z",
+     "iopub.status.busy": "2026-03-20T00:37:45.961328Z",
+     "iopub.status.idle": "2026-03-20T00:37:46.034559Z",
+     "shell.execute_reply": "2026-03-20T00:37:46.034127Z"
+    },
     "id": "TkkDHuhVEctz",
     "outputId": "5639d89b-13a5-4825-98e2-16e3c4139e5a"
    },
@@ -554,9 +674,18 @@
    "name": "python3"
   },
   "language_info": {
-   "name": "python"
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.15"
   }
  },
  "nbformat": 4,
  "nbformat_minor": 0
-}
\ No newline at end of file
+}
diff --git a/nbs/aliased_and_not_aliased_patch_extraction.ipynb b/nbs/aliased_and_not_aliased_patch_extraction.ipynb
index a76cd0e..7c524a8 100644
--- a/nbs/aliased_and_not_aliased_patch_extraction.ipynb
+++ b/nbs/aliased_and_not_aliased_patch_extraction.ipynb
@@ -43,10 +43,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:03.566822Z",
-     "iopub.status.busy": "2026-03-19T21:37:03.566357Z",
-     "iopub.status.idle": "2026-03-19T21:37:04.196898Z",
-     "shell.execute_reply": "2026-03-19T21:37:04.196535Z"
+     "iopub.execute_input": "2026-03-20T00:19:14.511078Z",
+     "iopub.status.busy": "2026-03-20T00:19:14.510851Z",
+     "iopub.status.idle": "2026-03-20T00:19:15.044569Z",
+     "shell.execute_reply": "2026-03-20T00:19:15.044155Z"
     }
    },
    "outputs": [],
@@ -83,10 +83,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:04.219178Z",
-     "iopub.status.busy": "2026-03-19T21:37:04.219010Z",
-     "iopub.status.idle": "2026-03-19T21:37:05.509416Z",
-     "shell.execute_reply": "2026-03-19T21:37:05.509107Z"
+     "iopub.execute_input": "2026-03-20T00:19:15.069687Z",
+     "iopub.status.busy": "2026-03-20T00:19:15.069514Z",
+     "iopub.status.idle": "2026-03-20T00:19:16.087411Z",
+     "shell.execute_reply": "2026-03-20T00:19:16.087085Z"
     }
    },
    "outputs": [],
@@ -106,10 +106,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:05.511569Z",
-     "iopub.status.busy": "2026-03-19T21:37:05.511367Z",
-     "iopub.status.idle": "2026-03-19T21:37:05.573370Z",
-     "shell.execute_reply": "2026-03-19T21:37:05.573027Z"
+     "iopub.execute_input": "2026-03-20T00:19:16.089097Z",
+     "iopub.status.busy": "2026-03-20T00:19:16.088943Z",
+     "iopub.status.idle": "2026-03-20T00:19:16.157385Z",
+     "shell.execute_reply": "2026-03-20T00:19:16.157031Z"
     }
    },
    "outputs": [],
@@ -127,10 +127,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:05.575009Z",
-     "iopub.status.busy": "2026-03-19T21:37:05.574898Z",
-     "iopub.status.idle": "2026-03-19T21:37:05.658175Z",
-     "shell.execute_reply": "2026-03-19T21:37:05.657895Z"
+     "iopub.execute_input": "2026-03-20T00:19:16.159174Z",
+     "iopub.status.busy": "2026-03-20T00:19:16.159044Z",
+     "iopub.status.idle": "2026-03-20T00:19:16.245571Z",
+     "shell.execute_reply": "2026-03-20T00:19:16.245288Z"
     }
    },
    "outputs": [],
@@ -155,10 +155,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:05.659631Z",
-     "iopub.status.busy": "2026-03-19T21:37:05.659520Z",
-     "iopub.status.idle": "2026-03-19T21:37:05.746269Z",
-     "shell.execute_reply": "2026-03-19T21:37:05.745932Z"
+     "iopub.execute_input": "2026-03-20T00:19:16.247101Z",
+     "iopub.status.busy": "2026-03-20T00:19:16.246985Z",
+     "iopub.status.idle": "2026-03-20T00:19:16.332104Z",
+     "shell.execute_reply": "2026-03-20T00:19:16.331845Z"
     }
    },
    "outputs": [],
@@ -194,10 +194,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:05.748034Z",
-     "iopub.status.busy": "2026-03-19T21:37:05.747912Z",
-     "iopub.status.idle": "2026-03-19T21:37:05.763255Z",
-     "shell.execute_reply": "2026-03-19T21:37:05.762907Z"
+     "iopub.execute_input": "2026-03-20T00:19:16.333584Z",
+     "iopub.status.busy": "2026-03-20T00:19:16.333480Z",
+     "iopub.status.idle": "2026-03-20T00:19:16.349135Z",
+     "shell.execute_reply": "2026-03-20T00:19:16.348801Z"
     }
    },
    "outputs": [],
@@ -244,10 +244,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:05.765237Z",
-     "iopub.status.busy": "2026-03-19T21:37:05.765084Z",
-     "iopub.status.idle": "2026-03-19T21:37:05.833477Z",
-     "shell.execute_reply": "2026-03-19T21:37:05.832925Z"
+     "iopub.execute_input": "2026-03-20T00:19:16.350769Z",
+     "iopub.status.busy": "2026-03-20T00:19:16.350659Z",
+     "iopub.status.idle": "2026-03-20T00:19:16.418756Z",
+     "shell.execute_reply": "2026-03-20T00:19:16.418248Z"
     }
    },
    "outputs": [],
@@ -268,10 +268,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:05.835763Z",
-     "iopub.status.busy": "2026-03-19T21:37:05.835593Z",
-     "iopub.status.idle": "2026-03-19T21:37:05.855579Z",
-     "shell.execute_reply": "2026-03-19T21:37:05.855224Z"
+     "iopub.execute_input": "2026-03-20T00:19:16.421014Z",
+     "iopub.status.busy": "2026-03-20T00:19:16.420857Z",
+     "iopub.status.idle": "2026-03-20T00:19:16.444108Z",
+     "shell.execute_reply": "2026-03-20T00:19:16.443661Z"
     }
    },
    "outputs": [],
diff --git a/nbs/canny.ipynb b/nbs/canny.ipynb
index 691c839..325d9c2 100644
--- a/nbs/canny.ipynb
+++ b/nbs/canny.ipynb
@@ -57,10 +57,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:07.936319Z",
-     "iopub.status.busy": "2026-03-19T21:37:07.935735Z",
-     "iopub.status.idle": "2026-03-19T21:37:08.754972Z",
-     "shell.execute_reply": "2026-03-19T21:37:08.754667Z"
+     "iopub.execute_input": "2026-03-20T00:19:18.543064Z",
+     "iopub.status.busy": "2026-03-20T00:19:18.542783Z",
+     "iopub.status.idle": "2026-03-20T00:19:19.347787Z",
+     "shell.execute_reply": "2026-03-20T00:19:19.347456Z"
     }
    },
    "outputs": [],
@@ -84,10 +84,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:08.772493Z",
-     "iopub.status.busy": "2026-03-19T21:37:08.772319Z",
-     "iopub.status.idle": "2026-03-19T21:37:09.287383Z",
-     "shell.execute_reply": "2026-03-19T21:37:09.286351Z"
+     "iopub.execute_input": "2026-03-20T00:19:19.365942Z",
+     "iopub.status.busy": "2026-03-20T00:19:19.365764Z",
+     "iopub.status.idle": "2026-03-20T00:19:19.906112Z",
+     "shell.execute_reply": "2026-03-20T00:19:19.905777Z"
     }
    },
    "outputs": [],
@@ -127,10 +127,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:09.291320Z",
-     "iopub.status.busy": "2026-03-19T21:37:09.291009Z",
-     "iopub.status.idle": "2026-03-19T21:37:09.545782Z",
-     "shell.execute_reply": "2026-03-19T21:37:09.545467Z"
+     "iopub.execute_input": "2026-03-20T00:19:19.908431Z",
+     "iopub.status.busy": "2026-03-20T00:19:19.908283Z",
+     "iopub.status.idle": "2026-03-20T00:19:20.149959Z",
+     "shell.execute_reply": "2026-03-20T00:19:20.149674Z"
     }
    },
    "outputs": [],
@@ -162,10 +162,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:09.547317Z",
-     "iopub.status.busy": "2026-03-19T21:37:09.547117Z",
-     "iopub.status.idle": "2026-03-19T21:37:09.784312Z",
-     "shell.execute_reply": "2026-03-19T21:37:09.784019Z"
+     "iopub.execute_input": "2026-03-20T00:19:20.151502Z",
+     "iopub.status.busy": "2026-03-20T00:19:20.151353Z",
+     "iopub.status.idle": "2026-03-20T00:19:20.385059Z",
+     "shell.execute_reply": "2026-03-20T00:19:20.384771Z"
     }
    },
    "outputs": [],
@@ -189,10 +189,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:09.785905Z",
-     "iopub.status.busy": "2026-03-19T21:37:09.785812Z",
-     "iopub.status.idle": "2026-03-19T21:37:09.888812Z",
-     "shell.execute_reply": "2026-03-19T21:37:09.888498Z"
+     "iopub.execute_input": "2026-03-20T00:19:20.386682Z",
+     "iopub.status.busy": "2026-03-20T00:19:20.386589Z",
+     "iopub.status.idle": "2026-03-20T00:19:20.481132Z",
+     "shell.execute_reply": "2026-03-20T00:19:20.480846Z"
     }
    },
    "outputs": [],
@@ -232,10 +232,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:09.890956Z",
-     "iopub.status.busy": "2026-03-19T21:37:09.890837Z",
-     "iopub.status.idle": "2026-03-19T21:37:10.081039Z",
-     "shell.execute_reply": "2026-03-19T21:37:10.080701Z"
+     "iopub.execute_input": "2026-03-20T00:19:20.483251Z",
+     "iopub.status.busy": "2026-03-20T00:19:20.483125Z",
+     "iopub.status.idle": "2026-03-20T00:19:20.653363Z",
+     "shell.execute_reply": "2026-03-20T00:19:20.653049Z"
     }
    },
    "outputs": [],
@@ -252,10 +252,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:10.082744Z",
-     "iopub.status.busy": "2026-03-19T21:37:10.082639Z",
-     "iopub.status.idle": "2026-03-19T21:37:10.180392Z",
-     "shell.execute_reply": "2026-03-19T21:37:10.180089Z"
+     "iopub.execute_input": "2026-03-20T00:19:20.654971Z",
+     "iopub.status.busy": "2026-03-20T00:19:20.654860Z",
+     "iopub.status.idle": "2026-03-20T00:19:20.746515Z",
+     "shell.execute_reply": "2026-03-20T00:19:20.746215Z"
     }
    },
    "outputs": [],
diff --git a/nbs/color_conversions.ipynb b/nbs/color_conversions.ipynb
index 5e41372..1ee1ebb 100644
--- a/nbs/color_conversions.ipynb
+++ b/nbs/color_conversions.ipynb
@@ -54,10 +54,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:37.597628Z",
-     "iopub.status.busy": "2026-03-19T21:25:37.597002Z",
-     "iopub.status.idle": "2026-03-19T21:25:38.226340Z",
-     "shell.execute_reply": "2026-03-19T21:25:38.220982Z"
+     "iopub.execute_input": "2026-03-20T00:19:22.818107Z",
+     "iopub.status.busy": "2026-03-20T00:19:22.817713Z",
+     "iopub.status.idle": "2026-03-20T00:19:23.380676Z",
+     "shell.execute_reply": "2026-03-20T00:19:23.380279Z"
     }
    },
    "outputs": [],
@@ -87,10 +87,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:38.247891Z",
-     "iopub.status.busy": "2026-03-19T21:25:38.247638Z",
-     "iopub.status.idle": "2026-03-19T21:25:40.176988Z",
-     "shell.execute_reply": "2026-03-19T21:25:40.176654Z"
+     "iopub.execute_input": "2026-03-20T00:19:23.406570Z",
+     "iopub.status.busy": "2026-03-20T00:19:23.406249Z",
+     "iopub.status.idle": "2026-03-20T00:19:25.146223Z",
+     "shell.execute_reply": "2026-03-20T00:19:25.145895Z"
     }
    },
    "outputs": [],
@@ -108,16 +108,16 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:40.178623Z",
-     "iopub.status.busy": "2026-03-19T21:25:40.178491Z",
-     "iopub.status.idle": "2026-03-19T21:25:40.214233Z",
-     "shell.execute_reply": "2026-03-19T21:25:40.213971Z"
+     "iopub.execute_input": "2026-03-20T00:19:25.148133Z",
+     "iopub.status.busy": "2026-03-20T00:19:25.147961Z",
+     "iopub.status.idle": "2026-03-20T00:19:25.189724Z",
+     "shell.execute_reply": "2026-03-20T00:19:25.189367Z"
     }
    },
    "outputs": [],
    "source": [
     "# read the image with Kornia\n",
-    "img_tensor = K.image_to_tensor(np.array(Image.open(\"simba.png\").convert(\"RGB\")))  # CxHxW\n",
+    "img_tensor = K.image_to_tensor(np.array(Image.open(\"simba.png\").convert(\"RGB\"))).float() / 255.0  # CxHxW\n",
     "img_array = K.tensor_to_image(img_tensor)\n",
     "\n",
     "plt.axis(\"off\")\n",
@@ -141,10 +141,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:40.215832Z",
-     "iopub.status.busy": "2026-03-19T21:25:40.215731Z",
-     "iopub.status.idle": "2026-03-19T21:25:40.246031Z",
-     "shell.execute_reply": "2026-03-19T21:25:40.245765Z"
+     "iopub.execute_input": "2026-03-20T00:19:25.191649Z",
+     "iopub.status.busy": "2026-03-20T00:19:25.191541Z",
+     "iopub.status.idle": "2026-03-20T00:19:25.227293Z",
+     "shell.execute_reply": "2026-03-20T00:19:25.227019Z"
     }
    },
    "outputs": [],
@@ -172,10 +172,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:40.247478Z",
-     "iopub.status.busy": "2026-03-19T21:25:40.247375Z",
-     "iopub.status.idle": "2026-03-19T21:25:40.283476Z",
-     "shell.execute_reply": "2026-03-19T21:25:40.283214Z"
+     "iopub.execute_input": "2026-03-20T00:19:25.229115Z",
+     "iopub.status.busy": "2026-03-20T00:19:25.228961Z",
+     "iopub.status.idle": "2026-03-20T00:19:25.267472Z",
+     "shell.execute_reply": "2026-03-20T00:19:25.267188Z"
     }
    },
    "outputs": [],
@@ -198,10 +198,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:40.285209Z",
-     "iopub.status.busy": "2026-03-19T21:25:40.285099Z",
-     "iopub.status.idle": "2026-03-19T21:25:40.320649Z",
-     "shell.execute_reply": "2026-03-19T21:25:40.320340Z"
+     "iopub.execute_input": "2026-03-20T00:19:25.269030Z",
+     "iopub.status.busy": "2026-03-20T00:19:25.268924Z",
+     "iopub.status.idle": "2026-03-20T00:19:25.306267Z",
+     "shell.execute_reply": "2026-03-20T00:19:25.305987Z"
     }
    },
    "outputs": [],
@@ -216,10 +216,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:40.322365Z",
-     "iopub.status.busy": "2026-03-19T21:25:40.322251Z",
-     "iopub.status.idle": "2026-03-19T21:25:40.359500Z",
-     "shell.execute_reply": "2026-03-19T21:25:40.359243Z"
+     "iopub.execute_input": "2026-03-20T00:19:25.307697Z",
+     "iopub.status.busy": "2026-03-20T00:19:25.307586Z",
+     "iopub.status.idle": "2026-03-20T00:19:25.349167Z",
+     "shell.execute_reply": "2026-03-20T00:19:25.348856Z"
     }
    },
    "outputs": [],
@@ -235,10 +235,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:40.360997Z",
-     "iopub.status.busy": "2026-03-19T21:25:40.360879Z",
-     "iopub.status.idle": "2026-03-19T21:25:40.405004Z",
-     "shell.execute_reply": "2026-03-19T21:25:40.404738Z"
+     "iopub.execute_input": "2026-03-20T00:19:25.350770Z",
+     "iopub.status.busy": "2026-03-20T00:19:25.350650Z",
+     "iopub.status.idle": "2026-03-20T00:19:25.396256Z",
+     "shell.execute_reply": "2026-03-20T00:19:25.395804Z"
     }
    },
    "outputs": [],
diff --git a/nbs/color_raw_to_rgb.ipynb b/nbs/color_raw_to_rgb.ipynb
index 31755c8..5d0e755 100644
--- a/nbs/color_raw_to_rgb.ipynb
+++ b/nbs/color_raw_to_rgb.ipynb
@@ -46,7 +46,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:37:50.801545Z",
+     "iopub.status.busy": "2026-03-20T00:37:50.800535Z",
+     "iopub.status.idle": "2026-03-20T00:37:54.514008Z",
+     "shell.execute_reply": "2026-03-20T00:37:54.513649Z"
+    }
+   },
    "outputs": [],
    "source": [
     "import io\n",
@@ -79,7 +86,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:37:54.544673Z",
+     "iopub.status.busy": "2026-03-20T00:37:54.544487Z",
+     "iopub.status.idle": "2026-03-20T00:37:56.723238Z",
+     "shell.execute_reply": "2026-03-20T00:37:56.722753Z"
+    }
+   },
    "outputs": [],
    "source": [
     "import kornia\n",
@@ -99,7 +113,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:37:56.725122Z",
+     "iopub.status.busy": "2026-03-20T00:37:56.724950Z",
+     "iopub.status.idle": "2026-03-20T00:37:56.742157Z",
+     "shell.execute_reply": "2026-03-20T00:37:56.741813Z"
+    }
+   },
    "outputs": [],
    "source": [
     "path = \"raw.dng\"\n",
@@ -134,7 +155,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:37:56.743861Z",
+     "iopub.status.busy": "2026-03-20T00:37:56.743729Z",
+     "iopub.status.idle": "2026-03-20T00:37:56.806227Z",
+     "shell.execute_reply": "2026-03-20T00:37:56.805854Z"
+    }
+   },
    "outputs": [],
    "source": [
     "# We find the data inside raw.raw_image\n",
@@ -156,7 +184,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:37:56.807997Z",
+     "iopub.status.busy": "2026-03-20T00:37:56.807874Z",
+     "iopub.status.idle": "2026-03-20T00:37:57.157203Z",
+     "shell.execute_reply": "2026-03-20T00:37:57.156904Z"
+    }
+   },
    "outputs": [],
    "source": [
     "npimg = np.moveaxis(np.squeeze((rgbtensor * 255.0).numpy().astype(np.uint8)), 0, 2)\n",
@@ -178,7 +213,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:37:57.159154Z",
+     "iopub.status.busy": "2026-03-20T00:37:57.159027Z",
+     "iopub.status.idle": "2026-03-20T00:37:57.514330Z",
+     "shell.execute_reply": "2026-03-20T00:37:57.514061Z"
+    }
+   },
    "outputs": [],
    "source": [
     "# if we do a pipeline were we first rotate the image, it will end up with a\n",
@@ -193,7 +235,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:37:57.515730Z",
+     "iopub.status.busy": "2026-03-20T00:37:57.515619Z",
+     "iopub.status.idle": "2026-03-20T00:37:57.615758Z",
+     "shell.execute_reply": "2026-03-20T00:37:57.615488Z"
+    }
+   },
    "outputs": [],
    "source": [
     "# If we crop, we can adjust for this by using a different cfa\n",
@@ -209,6 +258,18 @@
    "display_name": "python3",
    "language": "python",
    "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.15"
   }
  },
  "nbformat": 4,
diff --git a/nbs/color_yuv420_to_rgb.ipynb b/nbs/color_yuv420_to_rgb.ipynb
index 6ab3bd3..485a46f 100644
--- a/nbs/color_yuv420_to_rgb.ipynb
+++ b/nbs/color_yuv420_to_rgb.ipynb
@@ -46,7 +46,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:37:59.792023Z",
+     "iopub.status.busy": "2026-03-20T00:37:59.791540Z",
+     "iopub.status.idle": "2026-03-20T00:38:01.664883Z",
+     "shell.execute_reply": "2026-03-20T00:38:01.664438Z"
+    }
+   },
    "outputs": [],
    "source": [
     "import io\n",
@@ -79,7 +86,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:38:01.689542Z",
+     "iopub.status.busy": "2026-03-20T00:38:01.689378Z",
+     "iopub.status.idle": "2026-03-20T00:38:04.055503Z",
+     "shell.execute_reply": "2026-03-20T00:38:04.055134Z"
+    }
+   },
    "outputs": [],
    "source": [
     "import kornia\n",
@@ -103,7 +117,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:38:04.057435Z",
+     "iopub.status.busy": "2026-03-20T00:38:04.057248Z",
+     "iopub.status.idle": "2026-03-20T00:38:04.284712Z",
+     "shell.execute_reply": "2026-03-20T00:38:04.284362Z"
+    }
+   },
    "outputs": [],
    "source": [
     "import matplotlib.pyplot as plt\n",
@@ -136,7 +157,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:38:04.286440Z",
+     "iopub.status.busy": "2026-03-20T00:38:04.286299Z",
+     "iopub.status.idle": "2026-03-20T00:38:04.539055Z",
+     "shell.execute_reply": "2026-03-20T00:38:04.538666Z"
+    }
+   },
    "outputs": [],
    "source": [
     "y, uv = read_frame(\"foreman_qcif.yuv\", 0)  # using compression classic foreman\n",
@@ -163,7 +191,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:38:04.541037Z",
+     "iopub.status.busy": "2026-03-20T00:38:04.540909Z",
+     "iopub.status.idle": "2026-03-20T00:38:05.094746Z",
+     "shell.execute_reply": "2026-03-20T00:38:05.094474Z"
+    }
+   },
    "outputs": [],
    "source": [
     "import cv2\n",
@@ -196,6 +231,18 @@
    "display_name": "python3",
    "language": "python",
    "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.15"
   }
  },
  "nbformat": 4,
diff --git a/nbs/connected_components.ipynb b/nbs/connected_components.ipynb
index a3ff6f8..a71351f 100644
--- a/nbs/connected_components.ipynb
+++ b/nbs/connected_components.ipynb
@@ -43,10 +43,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:54.750298Z",
-     "iopub.status.busy": "2026-03-19T21:25:54.749508Z",
-     "iopub.status.idle": "2026-03-19T21:25:54.764472Z",
-     "shell.execute_reply": "2026-03-19T21:25:54.763659Z"
+     "iopub.execute_input": "2026-03-20T00:19:39.934449Z",
+     "iopub.status.busy": "2026-03-20T00:19:39.934332Z",
+     "iopub.status.idle": "2026-03-20T00:19:39.942322Z",
+     "shell.execute_reply": "2026-03-20T00:19:39.942014Z"
     }
    },
    "outputs": [],
@@ -59,10 +59,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:54.789087Z",
-     "iopub.status.busy": "2026-03-19T21:25:54.788895Z",
-     "iopub.status.idle": "2026-03-19T21:25:55.374778Z",
-     "shell.execute_reply": "2026-03-19T21:25:55.373529Z"
+     "iopub.execute_input": "2026-03-20T00:19:39.968969Z",
+     "iopub.status.busy": "2026-03-20T00:19:39.968618Z",
+     "iopub.status.idle": "2026-03-20T00:19:40.470826Z",
+     "shell.execute_reply": "2026-03-20T00:19:40.470147Z"
     }
    },
    "outputs": [],
@@ -92,10 +92,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:55.382776Z",
-     "iopub.status.busy": "2026-03-19T21:25:55.381399Z",
-     "iopub.status.idle": "2026-03-19T21:25:56.356254Z",
-     "shell.execute_reply": "2026-03-19T21:25:56.355983Z"
+     "iopub.execute_input": "2026-03-20T00:19:40.472954Z",
+     "iopub.status.busy": "2026-03-20T00:19:40.472788Z",
+     "iopub.status.idle": "2026-03-20T00:19:41.466209Z",
+     "shell.execute_reply": "2026-03-20T00:19:41.465813Z"
     }
    },
    "outputs": [],
@@ -119,10 +119,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:56.357857Z",
-     "iopub.status.busy": "2026-03-19T21:25:56.357725Z",
-     "iopub.status.idle": "2026-03-19T21:25:56.361563Z",
-     "shell.execute_reply": "2026-03-19T21:25:56.361317Z"
+     "iopub.execute_input": "2026-03-20T00:19:41.468555Z",
+     "iopub.status.busy": "2026-03-20T00:19:41.468249Z",
+     "iopub.status.idle": "2026-03-20T00:19:41.472385Z",
+     "shell.execute_reply": "2026-03-20T00:19:41.472115Z"
     }
    },
    "outputs": [],
@@ -181,10 +181,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:56.362938Z",
-     "iopub.status.busy": "2026-03-19T21:25:56.362820Z",
-     "iopub.status.idle": "2026-03-19T21:25:56.370013Z",
-     "shell.execute_reply": "2026-03-19T21:25:56.369761Z"
+     "iopub.execute_input": "2026-03-20T00:19:41.473759Z",
+     "iopub.status.busy": "2026-03-20T00:19:41.473657Z",
+     "iopub.status.idle": "2026-03-20T00:19:41.481430Z",
+     "shell.execute_reply": "2026-03-20T00:19:41.481179Z"
     }
    },
    "outputs": [],
@@ -211,10 +211,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:56.371372Z",
-     "iopub.status.busy": "2026-03-19T21:25:56.371287Z",
-     "iopub.status.idle": "2026-03-19T21:25:57.036715Z",
-     "shell.execute_reply": "2026-03-19T21:25:57.036409Z"
+     "iopub.execute_input": "2026-03-20T00:19:41.482829Z",
+     "iopub.status.busy": "2026-03-20T00:19:41.482732Z",
+     "iopub.status.idle": "2026-03-20T00:19:42.147478Z",
+     "shell.execute_reply": "2026-03-20T00:19:42.147186Z"
     }
    },
    "outputs": [],
@@ -228,10 +228,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:57.038204Z",
-     "iopub.status.busy": "2026-03-19T21:25:57.038105Z",
-     "iopub.status.idle": "2026-03-19T21:25:57.113881Z",
-     "shell.execute_reply": "2026-03-19T21:25:57.113610Z"
+     "iopub.execute_input": "2026-03-20T00:19:42.148903Z",
+     "iopub.status.busy": "2026-03-20T00:19:42.148810Z",
+     "iopub.status.idle": "2026-03-20T00:19:42.225336Z",
+     "shell.execute_reply": "2026-03-20T00:19:42.225043Z"
     }
    },
    "outputs": [],
@@ -251,10 +251,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:25:57.115388Z",
-     "iopub.status.busy": "2026-03-19T21:25:57.115269Z",
-     "iopub.status.idle": "2026-03-19T21:25:57.120091Z",
-     "shell.execute_reply": "2026-03-19T21:25:57.119812Z"
+     "iopub.execute_input": "2026-03-20T00:19:42.226767Z",
+     "iopub.status.busy": "2026-03-20T00:19:42.226652Z",
+     "iopub.status.idle": "2026-03-20T00:19:42.231133Z",
+     "shell.execute_reply": "2026-03-20T00:19:42.230865Z"
     }
    },
    "outputs": [],
diff --git a/nbs/data_augmentation_2d.ipynb b/nbs/data_augmentation_2d.ipynb
index 8b24088..8dba9ee 100644
--- a/nbs/data_augmentation_2d.ipynb
+++ b/nbs/data_augmentation_2d.ipynb
@@ -52,10 +52,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:12.314265Z",
-     "iopub.status.busy": "2026-03-19T21:37:12.313782Z",
-     "iopub.status.idle": "2026-03-19T21:37:13.322188Z",
-     "shell.execute_reply": "2026-03-19T21:37:13.321904Z"
+     "iopub.execute_input": "2026-03-20T00:19:44.325563Z",
+     "iopub.status.busy": "2026-03-20T00:19:44.325422Z",
+     "iopub.status.idle": "2026-03-20T00:19:45.339458Z",
+     "shell.execute_reply": "2026-03-20T00:19:45.339158Z"
     }
    },
    "outputs": [],
@@ -125,10 +125,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:13.323817Z",
-     "iopub.status.busy": "2026-03-19T21:37:13.323693Z",
-     "iopub.status.idle": "2026-03-19T21:37:13.639849Z",
-     "shell.execute_reply": "2026-03-19T21:37:13.639070Z"
+     "iopub.execute_input": "2026-03-20T00:19:45.341103Z",
+     "iopub.status.busy": "2026-03-20T00:19:45.340977Z",
+     "iopub.status.idle": "2026-03-20T00:19:45.445973Z",
+     "shell.execute_reply": "2026-03-20T00:19:45.445631Z"
     }
    },
    "outputs": [],
@@ -158,10 +158,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:13.672535Z",
-     "iopub.status.busy": "2026-03-19T21:37:13.672360Z",
-     "iopub.status.idle": "2026-03-19T21:37:13.685171Z",
-     "shell.execute_reply": "2026-03-19T21:37:13.684805Z"
+     "iopub.execute_input": "2026-03-20T00:19:45.466542Z",
+     "iopub.status.busy": "2026-03-20T00:19:45.466362Z",
+     "iopub.status.idle": "2026-03-20T00:19:45.479045Z",
+     "shell.execute_reply": "2026-03-20T00:19:45.478709Z"
     }
    },
    "outputs": [],
@@ -174,10 +174,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:13.687024Z",
-     "iopub.status.busy": "2026-03-19T21:37:13.686855Z",
-     "iopub.status.idle": "2026-03-19T21:37:13.689734Z",
-     "shell.execute_reply": "2026-03-19T21:37:13.689456Z"
+     "iopub.execute_input": "2026-03-20T00:19:45.480540Z",
+     "iopub.status.busy": "2026-03-20T00:19:45.480443Z",
+     "iopub.status.idle": "2026-03-20T00:19:45.482995Z",
+     "shell.execute_reply": "2026-03-20T00:19:45.482758Z"
     }
    },
    "outputs": [],
@@ -203,10 +203,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:13.691300Z",
-     "iopub.status.busy": "2026-03-19T21:37:13.691099Z",
-     "iopub.status.idle": "2026-03-19T21:37:13.791829Z",
-     "shell.execute_reply": "2026-03-19T21:37:13.791497Z"
+     "iopub.execute_input": "2026-03-20T00:19:45.484254Z",
+     "iopub.status.busy": "2026-03-20T00:19:45.484165Z",
+     "iopub.status.idle": "2026-03-20T00:19:45.574900Z",
+     "shell.execute_reply": "2026-03-20T00:19:45.574619Z"
     }
    },
    "outputs": [],
@@ -229,10 +229,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:13.794417Z",
-     "iopub.status.busy": "2026-03-19T21:37:13.794289Z",
-     "iopub.status.idle": "2026-03-19T21:37:13.798009Z",
-     "shell.execute_reply": "2026-03-19T21:37:13.797649Z"
+     "iopub.execute_input": "2026-03-20T00:19:45.578126Z",
+     "iopub.status.busy": "2026-03-20T00:19:45.578009Z",
+     "iopub.status.idle": "2026-03-20T00:19:45.581534Z",
+     "shell.execute_reply": "2026-03-20T00:19:45.581226Z"
     }
    },
    "outputs": [],
@@ -264,10 +264,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:13.799679Z",
-     "iopub.status.busy": "2026-03-19T21:37:13.799568Z",
-     "iopub.status.idle": "2026-03-19T21:37:13.905667Z",
-     "shell.execute_reply": "2026-03-19T21:37:13.905394Z"
+     "iopub.execute_input": "2026-03-20T00:19:45.583093Z",
+     "iopub.status.busy": "2026-03-20T00:19:45.583004Z",
+     "iopub.status.idle": "2026-03-20T00:19:45.689766Z",
+     "shell.execute_reply": "2026-03-20T00:19:45.689470Z"
     }
    },
    "outputs": [],
@@ -290,10 +290,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:13.907527Z",
-     "iopub.status.busy": "2026-03-19T21:37:13.907409Z",
-     "iopub.status.idle": "2026-03-19T21:37:14.032031Z",
-     "shell.execute_reply": "2026-03-19T21:37:14.031701Z"
+     "iopub.execute_input": "2026-03-20T00:19:45.691572Z",
+     "iopub.status.busy": "2026-03-20T00:19:45.691446Z",
+     "iopub.status.idle": "2026-03-20T00:19:45.817783Z",
+     "shell.execute_reply": "2026-03-20T00:19:45.817507Z"
     }
    },
    "outputs": [],
@@ -323,10 +323,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:14.033703Z",
-     "iopub.status.busy": "2026-03-19T21:37:14.033605Z",
-     "iopub.status.idle": "2026-03-19T21:37:14.181819Z",
-     "shell.execute_reply": "2026-03-19T21:37:14.181512Z"
+     "iopub.execute_input": "2026-03-20T00:19:45.819389Z",
+     "iopub.status.busy": "2026-03-20T00:19:45.819269Z",
+     "iopub.status.idle": "2026-03-20T00:19:45.982094Z",
+     "shell.execute_reply": "2026-03-20T00:19:45.981793Z"
     }
    },
    "outputs": [],
@@ -354,10 +354,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:14.183723Z",
-     "iopub.status.busy": "2026-03-19T21:37:14.183594Z",
-     "iopub.status.idle": "2026-03-19T21:37:14.320820Z",
-     "shell.execute_reply": "2026-03-19T21:37:14.320540Z"
+     "iopub.execute_input": "2026-03-20T00:19:45.984017Z",
+     "iopub.status.busy": "2026-03-20T00:19:45.983905Z",
+     "iopub.status.idle": "2026-03-20T00:19:46.123763Z",
+     "shell.execute_reply": "2026-03-20T00:19:46.123488Z"
     }
    },
    "outputs": [],
@@ -380,10 +380,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:14.322623Z",
-     "iopub.status.busy": "2026-03-19T21:37:14.322510Z",
-     "iopub.status.idle": "2026-03-19T21:37:14.480259Z",
-     "shell.execute_reply": "2026-03-19T21:37:14.479977Z"
+     "iopub.execute_input": "2026-03-20T00:19:46.125349Z",
+     "iopub.status.busy": "2026-03-20T00:19:46.125232Z",
+     "iopub.status.idle": "2026-03-20T00:19:46.302428Z",
+     "shell.execute_reply": "2026-03-20T00:19:46.302140Z"
     }
    },
    "outputs": [],
@@ -406,10 +406,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:14.481786Z",
-     "iopub.status.busy": "2026-03-19T21:37:14.481680Z",
-     "iopub.status.idle": "2026-03-19T21:37:14.611615Z",
-     "shell.execute_reply": "2026-03-19T21:37:14.611314Z"
+     "iopub.execute_input": "2026-03-20T00:19:46.304302Z",
+     "iopub.status.busy": "2026-03-20T00:19:46.304195Z",
+     "iopub.status.idle": "2026-03-20T00:19:46.433184Z",
+     "shell.execute_reply": "2026-03-20T00:19:46.432899Z"
     }
    },
    "outputs": [],
@@ -432,10 +432,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:14.613331Z",
-     "iopub.status.busy": "2026-03-19T21:37:14.613212Z",
-     "iopub.status.idle": "2026-03-19T21:37:14.748527Z",
-     "shell.execute_reply": "2026-03-19T21:37:14.748269Z"
+     "iopub.execute_input": "2026-03-20T00:19:46.434850Z",
+     "iopub.status.busy": "2026-03-20T00:19:46.434744Z",
+     "iopub.status.idle": "2026-03-20T00:19:46.597716Z",
+     "shell.execute_reply": "2026-03-20T00:19:46.597404Z"
     }
    },
    "outputs": [],
@@ -458,10 +458,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:14.749978Z",
-     "iopub.status.busy": "2026-03-19T21:37:14.749854Z",
-     "iopub.status.idle": "2026-03-19T21:37:14.848592Z",
-     "shell.execute_reply": "2026-03-19T21:37:14.848280Z"
+     "iopub.execute_input": "2026-03-20T00:19:46.599208Z",
+     "iopub.status.busy": "2026-03-20T00:19:46.599072Z",
+     "iopub.status.idle": "2026-03-20T00:19:46.699187Z",
+     "shell.execute_reply": "2026-03-20T00:19:46.698914Z"
     }
    },
    "outputs": [],
@@ -484,10 +484,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:14.850336Z",
-     "iopub.status.busy": "2026-03-19T21:37:14.850230Z",
-     "iopub.status.idle": "2026-03-19T21:37:14.949345Z",
-     "shell.execute_reply": "2026-03-19T21:37:14.949076Z"
+     "iopub.execute_input": "2026-03-20T00:19:46.700912Z",
+     "iopub.status.busy": "2026-03-20T00:19:46.700809Z",
+     "iopub.status.idle": "2026-03-20T00:19:46.801692Z",
+     "shell.execute_reply": "2026-03-20T00:19:46.801434Z"
     }
    },
    "outputs": [],
@@ -510,10 +510,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:14.951009Z",
-     "iopub.status.busy": "2026-03-19T21:37:14.950900Z",
-     "iopub.status.idle": "2026-03-19T21:37:15.050787Z",
-     "shell.execute_reply": "2026-03-19T21:37:15.050490Z"
+     "iopub.execute_input": "2026-03-20T00:19:46.803687Z",
+     "iopub.status.busy": "2026-03-20T00:19:46.803570Z",
+     "iopub.status.idle": "2026-03-20T00:19:46.901200Z",
+     "shell.execute_reply": "2026-03-20T00:19:46.900933Z"
     }
    },
    "outputs": [],
@@ -536,10 +536,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:15.052593Z",
-     "iopub.status.busy": "2026-03-19T21:37:15.052471Z",
-     "iopub.status.idle": "2026-03-19T21:37:15.217259Z",
-     "shell.execute_reply": "2026-03-19T21:37:15.216983Z"
+     "iopub.execute_input": "2026-03-20T00:19:46.902920Z",
+     "iopub.status.busy": "2026-03-20T00:19:46.902813Z",
+     "iopub.status.idle": "2026-03-20T00:19:47.069013Z",
+     "shell.execute_reply": "2026-03-20T00:19:47.068723Z"
     }
    },
    "outputs": [],
@@ -562,10 +562,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:15.218794Z",
-     "iopub.status.busy": "2026-03-19T21:37:15.218679Z",
-     "iopub.status.idle": "2026-03-19T21:37:15.317431Z",
-     "shell.execute_reply": "2026-03-19T21:37:15.317116Z"
+     "iopub.execute_input": "2026-03-20T00:19:47.070443Z",
+     "iopub.status.busy": "2026-03-20T00:19:47.070339Z",
+     "iopub.status.idle": "2026-03-20T00:19:47.169070Z",
+     "shell.execute_reply": "2026-03-20T00:19:47.168705Z"
     }
    },
    "outputs": [],
@@ -588,10 +588,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:15.319315Z",
-     "iopub.status.busy": "2026-03-19T21:37:15.319149Z",
-     "iopub.status.idle": "2026-03-19T21:37:15.417286Z",
-     "shell.execute_reply": "2026-03-19T21:37:15.416980Z"
+     "iopub.execute_input": "2026-03-20T00:19:47.170551Z",
+     "iopub.status.busy": "2026-03-20T00:19:47.170459Z",
+     "iopub.status.idle": "2026-03-20T00:19:47.269653Z",
+     "shell.execute_reply": "2026-03-20T00:19:47.269330Z"
     }
    },
    "outputs": [],
@@ -614,10 +614,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:15.418789Z",
-     "iopub.status.busy": "2026-03-19T21:37:15.418687Z",
-     "iopub.status.idle": "2026-03-19T21:37:15.563574Z",
-     "shell.execute_reply": "2026-03-19T21:37:15.563219Z"
+     "iopub.execute_input": "2026-03-20T00:19:47.271133Z",
+     "iopub.status.busy": "2026-03-20T00:19:47.271011Z",
+     "iopub.status.idle": "2026-03-20T00:19:47.418576Z",
+     "shell.execute_reply": "2026-03-20T00:19:47.418254Z"
     }
    },
    "outputs": [],
@@ -640,10 +640,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:15.565340Z",
-     "iopub.status.busy": "2026-03-19T21:37:15.565238Z",
-     "iopub.status.idle": "2026-03-19T21:37:15.704587Z",
-     "shell.execute_reply": "2026-03-19T21:37:15.704268Z"
+     "iopub.execute_input": "2026-03-20T00:19:47.421502Z",
+     "iopub.status.busy": "2026-03-20T00:19:47.421387Z",
+     "iopub.status.idle": "2026-03-20T00:19:47.561676Z",
+     "shell.execute_reply": "2026-03-20T00:19:47.561338Z"
     }
    },
    "outputs": [],
@@ -666,10 +666,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:15.707065Z",
-     "iopub.status.busy": "2026-03-19T21:37:15.706969Z",
-     "iopub.status.idle": "2026-03-19T21:37:15.832557Z",
-     "shell.execute_reply": "2026-03-19T21:37:15.832271Z"
+     "iopub.execute_input": "2026-03-20T00:19:47.564193Z",
+     "iopub.status.busy": "2026-03-20T00:19:47.564102Z",
+     "iopub.status.idle": "2026-03-20T00:19:47.693138Z",
+     "shell.execute_reply": "2026-03-20T00:19:47.692866Z"
     }
    },
    "outputs": [],
@@ -692,10 +692,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:15.834397Z",
-     "iopub.status.busy": "2026-03-19T21:37:15.834290Z",
-     "iopub.status.idle": "2026-03-19T21:37:15.981505Z",
-     "shell.execute_reply": "2026-03-19T21:37:15.981227Z"
+     "iopub.execute_input": "2026-03-20T00:19:47.694966Z",
+     "iopub.status.busy": "2026-03-20T00:19:47.694845Z",
+     "iopub.status.idle": "2026-03-20T00:19:47.854308Z",
+     "shell.execute_reply": "2026-03-20T00:19:47.854037Z"
     }
    },
    "outputs": [],
@@ -718,10 +718,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:15.983048Z",
-     "iopub.status.busy": "2026-03-19T21:37:15.982937Z",
-     "iopub.status.idle": "2026-03-19T21:37:16.089378Z",
-     "shell.execute_reply": "2026-03-19T21:37:16.089102Z"
+     "iopub.execute_input": "2026-03-20T00:19:47.856112Z",
+     "iopub.status.busy": "2026-03-20T00:19:47.856006Z",
+     "iopub.status.idle": "2026-03-20T00:19:47.959635Z",
+     "shell.execute_reply": "2026-03-20T00:19:47.959347Z"
     }
    },
    "outputs": [],
@@ -744,10 +744,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:16.091289Z",
-     "iopub.status.busy": "2026-03-19T21:37:16.091087Z",
-     "iopub.status.idle": "2026-03-19T21:37:16.189158Z",
-     "shell.execute_reply": "2026-03-19T21:37:16.188879Z"
+     "iopub.execute_input": "2026-03-20T00:19:47.961514Z",
+     "iopub.status.busy": "2026-03-20T00:19:47.961396Z",
+     "iopub.status.idle": "2026-03-20T00:19:48.059943Z",
+     "shell.execute_reply": "2026-03-20T00:19:48.059668Z"
     }
    },
    "outputs": [],
@@ -776,10 +776,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:16.190858Z",
-     "iopub.status.busy": "2026-03-19T21:37:16.190740Z",
-     "iopub.status.idle": "2026-03-19T21:37:16.310323Z",
-     "shell.execute_reply": "2026-03-19T21:37:16.310044Z"
+     "iopub.execute_input": "2026-03-20T00:19:48.061632Z",
+     "iopub.status.busy": "2026-03-20T00:19:48.061519Z",
+     "iopub.status.idle": "2026-03-20T00:19:48.203960Z",
+     "shell.execute_reply": "2026-03-20T00:19:48.203686Z"
     }
    },
    "outputs": [],
@@ -802,10 +802,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:16.311720Z",
-     "iopub.status.busy": "2026-03-19T21:37:16.311608Z",
-     "iopub.status.idle": "2026-03-19T21:37:16.416085Z",
-     "shell.execute_reply": "2026-03-19T21:37:16.415795Z"
+     "iopub.execute_input": "2026-03-20T00:19:48.205437Z",
+     "iopub.status.busy": "2026-03-20T00:19:48.205317Z",
+     "iopub.status.idle": "2026-03-20T00:19:48.316805Z",
+     "shell.execute_reply": "2026-03-20T00:19:48.316534Z"
     }
    },
    "outputs": [],
@@ -828,10 +828,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:16.417660Z",
-     "iopub.status.busy": "2026-03-19T21:37:16.417544Z",
-     "iopub.status.idle": "2026-03-19T21:37:16.518362Z",
-     "shell.execute_reply": "2026-03-19T21:37:16.518072Z"
+     "iopub.execute_input": "2026-03-20T00:19:48.318461Z",
+     "iopub.status.busy": "2026-03-20T00:19:48.318352Z",
+     "iopub.status.idle": "2026-03-20T00:19:48.420857Z",
+     "shell.execute_reply": "2026-03-20T00:19:48.420551Z"
     }
    },
    "outputs": [],
@@ -862,10 +862,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:16.520371Z",
-     "iopub.status.busy": "2026-03-19T21:37:16.520258Z",
-     "iopub.status.idle": "2026-03-19T21:37:16.612313Z",
-     "shell.execute_reply": "2026-03-19T21:37:16.611899Z"
+     "iopub.execute_input": "2026-03-20T00:19:48.422630Z",
+     "iopub.status.busy": "2026-03-20T00:19:48.422514Z",
+     "iopub.status.idle": "2026-03-20T00:19:48.511982Z",
+     "shell.execute_reply": "2026-03-20T00:19:48.511700Z"
     }
    },
    "outputs": [],
@@ -895,10 +895,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:16.614862Z",
-     "iopub.status.busy": "2026-03-19T21:37:16.614612Z",
-     "iopub.status.idle": "2026-03-19T21:37:16.716622Z",
-     "shell.execute_reply": "2026-03-19T21:37:16.716338Z"
+     "iopub.execute_input": "2026-03-20T00:19:48.513480Z",
+     "iopub.status.busy": "2026-03-20T00:19:48.513368Z",
+     "iopub.status.idle": "2026-03-20T00:19:48.612730Z",
+     "shell.execute_reply": "2026-03-20T00:19:48.612418Z"
     }
    },
    "outputs": [],
@@ -921,10 +921,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:16.718412Z",
-     "iopub.status.busy": "2026-03-19T21:37:16.718263Z",
-     "iopub.status.idle": "2026-03-19T21:37:16.829645Z",
-     "shell.execute_reply": "2026-03-19T21:37:16.829345Z"
+     "iopub.execute_input": "2026-03-20T00:19:48.614428Z",
+     "iopub.status.busy": "2026-03-20T00:19:48.614313Z",
+     "iopub.status.idle": "2026-03-20T00:19:48.725996Z",
+     "shell.execute_reply": "2026-03-20T00:19:48.725695Z"
     }
    },
    "outputs": [],
@@ -957,10 +957,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:16.831537Z",
-     "iopub.status.busy": "2026-03-19T21:37:16.831428Z",
-     "iopub.status.idle": "2026-03-19T21:37:16.929238Z",
-     "shell.execute_reply": "2026-03-19T21:37:16.928918Z"
+     "iopub.execute_input": "2026-03-20T00:19:48.727765Z",
+     "iopub.status.busy": "2026-03-20T00:19:48.727658Z",
+     "iopub.status.idle": "2026-03-20T00:19:48.829327Z",
+     "shell.execute_reply": "2026-03-20T00:19:48.829036Z"
     }
    },
    "outputs": [],
@@ -995,10 +995,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:16.931217Z",
-     "iopub.status.busy": "2026-03-19T21:37:16.931075Z",
-     "iopub.status.idle": "2026-03-19T21:37:17.095460Z",
-     "shell.execute_reply": "2026-03-19T21:37:17.095165Z"
+     "iopub.execute_input": "2026-03-20T00:19:48.831050Z",
+     "iopub.status.busy": "2026-03-20T00:19:48.830933Z",
+     "iopub.status.idle": "2026-03-20T00:19:48.997573Z",
+     "shell.execute_reply": "2026-03-20T00:19:48.997206Z"
     }
    },
    "outputs": [],
@@ -1028,10 +1028,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:17.097149Z",
-     "iopub.status.busy": "2026-03-19T21:37:17.097016Z",
-     "iopub.status.idle": "2026-03-19T21:37:18.134605Z",
-     "shell.execute_reply": "2026-03-19T21:37:18.134317Z"
+     "iopub.execute_input": "2026-03-20T00:19:48.999033Z",
+     "iopub.status.busy": "2026-03-20T00:19:48.998919Z",
+     "iopub.status.idle": "2026-03-20T00:19:50.169805Z",
+     "shell.execute_reply": "2026-03-20T00:19:50.169533Z"
     }
    },
    "outputs": [],
@@ -1063,10 +1063,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:18.136667Z",
-     "iopub.status.busy": "2026-03-19T21:37:18.136564Z",
-     "iopub.status.idle": "2026-03-19T21:37:18.240966Z",
-     "shell.execute_reply": "2026-03-19T21:37:18.240689Z"
+     "iopub.execute_input": "2026-03-20T00:19:50.171831Z",
+     "iopub.status.busy": "2026-03-20T00:19:50.171720Z",
+     "iopub.status.idle": "2026-03-20T00:19:50.281675Z",
+     "shell.execute_reply": "2026-03-20T00:19:50.281230Z"
     }
    },
    "outputs": [],
@@ -1091,10 +1091,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:18.242347Z",
-     "iopub.status.busy": "2026-03-19T21:37:18.242233Z",
-     "iopub.status.idle": "2026-03-19T21:37:18.335011Z",
-     "shell.execute_reply": "2026-03-19T21:37:18.334746Z"
+     "iopub.execute_input": "2026-03-20T00:19:50.283469Z",
+     "iopub.status.busy": "2026-03-20T00:19:50.283332Z",
+     "iopub.status.idle": "2026-03-20T00:19:50.376199Z",
+     "shell.execute_reply": "2026-03-20T00:19:50.375895Z"
     }
    },
    "outputs": [],
@@ -1117,10 +1117,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:18.336528Z",
-     "iopub.status.busy": "2026-03-19T21:37:18.336399Z",
-     "iopub.status.idle": "2026-03-19T21:37:18.431369Z",
-     "shell.execute_reply": "2026-03-19T21:37:18.431093Z"
+     "iopub.execute_input": "2026-03-20T00:19:50.377667Z",
+     "iopub.status.busy": "2026-03-20T00:19:50.377539Z",
+     "iopub.status.idle": "2026-03-20T00:19:50.475221Z",
+     "shell.execute_reply": "2026-03-20T00:19:50.474927Z"
     }
    },
    "outputs": [],
@@ -1143,10 +1143,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:18.432863Z",
-     "iopub.status.busy": "2026-03-19T21:37:18.432750Z",
-     "iopub.status.idle": "2026-03-19T21:37:18.554215Z",
-     "shell.execute_reply": "2026-03-19T21:37:18.553949Z"
+     "iopub.execute_input": "2026-03-20T00:19:50.476691Z",
+     "iopub.status.busy": "2026-03-20T00:19:50.476571Z",
+     "iopub.status.idle": "2026-03-20T00:19:50.610022Z",
+     "shell.execute_reply": "2026-03-20T00:19:50.609622Z"
     }
    },
    "outputs": [],
@@ -1169,10 +1169,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:18.555784Z",
-     "iopub.status.busy": "2026-03-19T21:37:18.555671Z",
-     "iopub.status.idle": "2026-03-19T21:37:18.659023Z",
-     "shell.execute_reply": "2026-03-19T21:37:18.658726Z"
+     "iopub.execute_input": "2026-03-20T00:19:50.611667Z",
+     "iopub.status.busy": "2026-03-20T00:19:50.611541Z",
+     "iopub.status.idle": "2026-03-20T00:19:50.719130Z",
+     "shell.execute_reply": "2026-03-20T00:19:50.718810Z"
     }
    },
    "outputs": [],
@@ -1205,10 +1205,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:18.661026Z",
-     "iopub.status.busy": "2026-03-19T21:37:18.660920Z",
-     "iopub.status.idle": "2026-03-19T21:37:18.768877Z",
-     "shell.execute_reply": "2026-03-19T21:37:18.768580Z"
+     "iopub.execute_input": "2026-03-20T00:19:50.721342Z",
+     "iopub.status.busy": "2026-03-20T00:19:50.721248Z",
+     "iopub.status.idle": "2026-03-20T00:19:50.830760Z",
+     "shell.execute_reply": "2026-03-20T00:19:50.830463Z"
     }
    },
    "outputs": [],
@@ -1231,10 +1231,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:18.770693Z",
-     "iopub.status.busy": "2026-03-19T21:37:18.770573Z",
-     "iopub.status.idle": "2026-03-19T21:37:18.867165Z",
-     "shell.execute_reply": "2026-03-19T21:37:18.866896Z"
+     "iopub.execute_input": "2026-03-20T00:19:50.832590Z",
+     "iopub.status.busy": "2026-03-20T00:19:50.832472Z",
+     "iopub.status.idle": "2026-03-20T00:19:50.931379Z",
+     "shell.execute_reply": "2026-03-20T00:19:50.931018Z"
     }
    },
    "outputs": [],
@@ -1257,10 +1257,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:18.868653Z",
-     "iopub.status.busy": "2026-03-19T21:37:18.868540Z",
-     "iopub.status.idle": "2026-03-19T21:37:18.961858Z",
-     "shell.execute_reply": "2026-03-19T21:37:18.961568Z"
+     "iopub.execute_input": "2026-03-20T00:19:50.933477Z",
+     "iopub.status.busy": "2026-03-20T00:19:50.933353Z",
+     "iopub.status.idle": "2026-03-20T00:19:51.035260Z",
+     "shell.execute_reply": "2026-03-20T00:19:51.034901Z"
     }
    },
    "outputs": [],
@@ -1291,10 +1291,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:18.963370Z",
-     "iopub.status.busy": "2026-03-19T21:37:18.963254Z",
-     "iopub.status.idle": "2026-03-19T21:37:19.093135Z",
-     "shell.execute_reply": "2026-03-19T21:37:19.092852Z"
+     "iopub.execute_input": "2026-03-20T00:19:51.036944Z",
+     "iopub.status.busy": "2026-03-20T00:19:51.036792Z",
+     "iopub.status.idle": "2026-03-20T00:19:51.194127Z",
+     "shell.execute_reply": "2026-03-20T00:19:51.193819Z"
     }
    },
    "outputs": [],
@@ -1317,10 +1317,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:19.094610Z",
-     "iopub.status.busy": "2026-03-19T21:37:19.094496Z",
-     "iopub.status.idle": "2026-03-19T21:37:19.192649Z",
-     "shell.execute_reply": "2026-03-19T21:37:19.192351Z"
+     "iopub.execute_input": "2026-03-20T00:19:51.195629Z",
+     "iopub.status.busy": "2026-03-20T00:19:51.195510Z",
+     "iopub.status.idle": "2026-03-20T00:19:51.292580Z",
+     "shell.execute_reply": "2026-03-20T00:19:51.292245Z"
     }
    },
    "outputs": [],
@@ -1343,10 +1343,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:19.194137Z",
-     "iopub.status.busy": "2026-03-19T21:37:19.194021Z",
-     "iopub.status.idle": "2026-03-19T21:37:19.335834Z",
-     "shell.execute_reply": "2026-03-19T21:37:19.335545Z"
+     "iopub.execute_input": "2026-03-20T00:19:51.294128Z",
+     "iopub.status.busy": "2026-03-20T00:19:51.294006Z",
+     "iopub.status.idle": "2026-03-20T00:19:51.445048Z",
+     "shell.execute_reply": "2026-03-20T00:19:51.444724Z"
     }
    },
    "outputs": [],
@@ -1378,10 +1378,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:19.337426Z",
-     "iopub.status.busy": "2026-03-19T21:37:19.337310Z",
-     "iopub.status.idle": "2026-03-19T21:37:19.338827Z",
-     "shell.execute_reply": "2026-03-19T21:37:19.338587Z"
+     "iopub.execute_input": "2026-03-20T00:19:51.447094Z",
+     "iopub.status.busy": "2026-03-20T00:19:51.446959Z",
+     "iopub.status.idle": "2026-03-20T00:19:51.448869Z",
+     "shell.execute_reply": "2026-03-20T00:19:51.448517Z"
     }
    },
    "outputs": [],
diff --git a/nbs/data_augmentation_kornia_lightning.ipynb b/nbs/data_augmentation_kornia_lightning.ipynb
index 9881e89..544efff 100644
--- a/nbs/data_augmentation_kornia_lightning.ipynb
+++ b/nbs/data_augmentation_kornia_lightning.ipynb
@@ -56,7 +56,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T01:14:32.974778Z",
+     "iopub.status.busy": "2026-03-20T01:14:32.974590Z",
+     "iopub.status.idle": "2026-03-20T01:14:36.801131Z",
+     "shell.execute_reply": "2026-03-20T01:14:36.800771Z"
+    }
+   },
    "outputs": [],
    "source": [
     "import os\n",
@@ -83,7 +90,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T01:14:36.802818Z",
+     "iopub.status.busy": "2026-03-20T01:14:36.802645Z",
+     "iopub.status.idle": "2026-03-20T01:14:36.805655Z",
+     "shell.execute_reply": "2026-03-20T01:14:36.805334Z"
+    }
+   },
    "outputs": [],
    "source": [
     "class DataAugmentation(nn.Module):\n",
@@ -120,7 +134,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T01:14:36.807406Z",
+     "iopub.status.busy": "2026-03-20T01:14:36.807251Z",
+     "iopub.status.idle": "2026-03-20T01:14:36.809658Z",
+     "shell.execute_reply": "2026-03-20T01:14:36.809385Z"
+    }
+   },
    "outputs": [],
    "source": [
     "class PreProcess(nn.Module):\n",
@@ -146,7 +167,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T01:14:36.811277Z",
+     "iopub.status.busy": "2026-03-20T01:14:36.811146Z",
+     "iopub.status.idle": "2026-03-20T01:14:36.816440Z",
+     "shell.execute_reply": "2026-03-20T01:14:36.816154Z"
+    }
+   },
    "outputs": [],
    "source": [
     "class CoolSystem(pl.LightningModule):\n",
@@ -202,17 +230,17 @@
     "    def train_dataloader(self):\n",
     "        # REQUIRED\n",
     "        dataset = CIFAR10(os.getcwd(), train=True, download=False, transform=self.preprocess)\n",
-    "        loader = DataLoader(dataset, batch_size=32, num_workers=1)\n",
+    "        loader = DataLoader(dataset, batch_size=32, num_workers=0)\n",
     "        return loader\n",
     "\n",
     "    def val_dataloader(self):\n",
     "        dataset = CIFAR10(os.getcwd(), train=True, download=False, transform=self.preprocess)\n",
-    "        loader = DataLoader(dataset, batch_size=32, num_workers=1)\n",
+    "        loader = DataLoader(dataset, batch_size=32, num_workers=0)\n",
     "        return loader\n",
     "\n",
     "    def test_dataloader(self):\n",
     "        dataset = CIFAR10(os.getcwd(), train=False, download=False, transform=self.preprocess)\n",
-    "        loader = DataLoader(dataset, batch_size=16, num_workers=1)\n",
+    "        loader = DataLoader(dataset, batch_size=16, num_workers=0)\n",
     "        return loader"
    ]
   },
@@ -226,7 +254,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T01:14:36.818478Z",
+     "iopub.status.busy": "2026-03-20T01:14:36.818257Z",
+     "iopub.status.idle": "2026-03-20T01:14:47.782088Z",
+     "shell.execute_reply": "2026-03-20T01:14:47.781571Z"
+    }
+   },
    "outputs": [],
    "source": [
     "from pytorch_lightning import Trainer\n",
@@ -253,7 +288,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T01:14:47.784329Z",
+     "iopub.status.busy": "2026-03-20T01:14:47.784120Z",
+     "iopub.status.idle": "2026-03-20T01:14:50.417346Z",
+     "shell.execute_reply": "2026-03-20T01:14:50.417062Z"
+    }
+   },
    "outputs": [],
    "source": [
     "trainer.test(model)"
@@ -269,7 +311,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T01:14:50.418845Z",
+     "iopub.status.busy": "2026-03-20T01:14:50.418741Z",
+     "iopub.status.idle": "2026-03-20T01:14:50.420311Z",
+     "shell.execute_reply": "2026-03-20T01:14:50.420074Z"
+    }
+   },
    "outputs": [],
    "source": [
     "# # Start tensorboard.\n",
@@ -283,6 +332,18 @@
    "display_name": "python3",
    "language": "python",
    "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.15"
   }
  },
  "nbformat": 4,
diff --git a/nbs/data_augmentation_mosiac.ipynb b/nbs/data_augmentation_mosiac.ipynb
index a9de317..9eb901c 100644
--- a/nbs/data_augmentation_mosiac.ipynb
+++ b/nbs/data_augmentation_mosiac.ipynb
@@ -52,10 +52,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:21.606370Z",
-     "iopub.status.busy": "2026-03-19T21:37:21.605880Z",
-     "iopub.status.idle": "2026-03-19T21:37:21.756681Z",
-     "shell.execute_reply": "2026-03-19T21:37:21.756364Z"
+     "iopub.execute_input": "2026-03-20T00:19:56.794189Z",
+     "iopub.status.busy": "2026-03-20T00:19:56.793999Z",
+     "iopub.status.idle": "2026-03-20T00:19:56.934439Z",
+     "shell.execute_reply": "2026-03-20T00:19:56.933982Z"
     }
    },
    "outputs": [],
@@ -85,10 +85,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:21.781843Z",
-     "iopub.status.busy": "2026-03-19T21:37:21.781654Z",
-     "iopub.status.idle": "2026-03-19T21:37:22.854879Z",
-     "shell.execute_reply": "2026-03-19T21:37:22.854518Z"
+     "iopub.execute_input": "2026-03-20T00:19:56.956325Z",
+     "iopub.status.busy": "2026-03-20T00:19:56.956159Z",
+     "iopub.status.idle": "2026-03-20T00:19:57.966150Z",
+     "shell.execute_reply": "2026-03-20T00:19:57.965646Z"
     }
    },
    "outputs": [],
@@ -113,10 +113,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:22.856868Z",
-     "iopub.status.busy": "2026-03-19T21:37:22.856647Z",
-     "iopub.status.idle": "2026-03-19T21:37:23.019071Z",
-     "shell.execute_reply": "2026-03-19T21:37:23.018791Z"
+     "iopub.execute_input": "2026-03-20T00:19:57.968024Z",
+     "iopub.status.busy": "2026-03-20T00:19:57.967830Z",
+     "iopub.status.idle": "2026-03-20T00:19:58.128147Z",
+     "shell.execute_reply": "2026-03-20T00:19:58.127855Z"
     }
    },
    "outputs": [],
@@ -136,10 +136,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:23.022673Z",
-     "iopub.status.busy": "2026-03-19T21:37:23.022559Z",
-     "iopub.status.idle": "2026-03-19T21:37:23.127299Z",
-     "shell.execute_reply": "2026-03-19T21:37:23.126951Z"
+     "iopub.execute_input": "2026-03-20T00:19:58.134261Z",
+     "iopub.status.busy": "2026-03-20T00:19:58.134104Z",
+     "iopub.status.idle": "2026-03-20T00:19:58.237796Z",
+     "shell.execute_reply": "2026-03-20T00:19:58.237493Z"
     }
    },
    "outputs": [],
diff --git a/nbs/data_augmentation_segmentation.ipynb b/nbs/data_augmentation_segmentation.ipynb
index 6844cda..341a577 100644
--- a/nbs/data_augmentation_segmentation.ipynb
+++ b/nbs/data_augmentation_segmentation.ipynb
@@ -54,10 +54,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:08.847484Z",
-     "iopub.status.busy": "2026-03-19T21:26:08.846997Z",
-     "iopub.status.idle": "2026-03-19T21:26:10.072843Z",
-     "shell.execute_reply": "2026-03-19T21:26:10.071280Z"
+     "iopub.execute_input": "2026-03-20T00:20:00.394691Z",
+     "iopub.status.busy": "2026-03-20T00:20:00.394521Z",
+     "iopub.status.idle": "2026-03-20T00:20:00.997757Z",
+     "shell.execute_reply": "2026-03-20T00:20:00.997235Z"
     }
    },
    "outputs": [],
@@ -87,10 +87,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:10.106588Z",
-     "iopub.status.busy": "2026-03-19T21:26:10.106246Z",
-     "iopub.status.idle": "2026-03-19T21:26:11.155656Z",
-     "shell.execute_reply": "2026-03-19T21:26:11.155209Z"
+     "iopub.execute_input": "2026-03-20T00:20:01.023273Z",
+     "iopub.status.busy": "2026-03-20T00:20:01.022946Z",
+     "iopub.status.idle": "2026-03-20T00:20:02.063672Z",
+     "shell.execute_reply": "2026-03-20T00:20:02.063364Z"
     }
    },
    "outputs": [],
@@ -119,10 +119,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:11.157454Z",
-     "iopub.status.busy": "2026-03-19T21:26:11.157257Z",
-     "iopub.status.idle": "2026-03-19T21:26:11.159904Z",
-     "shell.execute_reply": "2026-03-19T21:26:11.159676Z"
+     "iopub.execute_input": "2026-03-20T00:20:02.065450Z",
+     "iopub.status.busy": "2026-03-20T00:20:02.065312Z",
+     "iopub.status.idle": "2026-03-20T00:20:02.068207Z",
+     "shell.execute_reply": "2026-03-20T00:20:02.067959Z"
     }
    },
    "outputs": [],
@@ -158,10 +158,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:11.161189Z",
-     "iopub.status.busy": "2026-03-19T21:26:11.161082Z",
-     "iopub.status.idle": "2026-03-19T21:26:11.570399Z",
-     "shell.execute_reply": "2026-03-19T21:26:11.570004Z"
+     "iopub.execute_input": "2026-03-20T00:20:02.069648Z",
+     "iopub.status.busy": "2026-03-20T00:20:02.069546Z",
+     "iopub.status.idle": "2026-03-20T00:20:02.458638Z",
+     "shell.execute_reply": "2026-03-20T00:20:02.458331Z"
     }
    },
    "outputs": [],
diff --git a/nbs/data_augmentation_sequential.ipynb b/nbs/data_augmentation_sequential.ipynb
index a633ee1..8f4eb71 100644
--- a/nbs/data_augmentation_sequential.ipynb
+++ b/nbs/data_augmentation_sequential.ipynb
@@ -50,10 +50,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:25.239712Z",
-     "iopub.status.busy": "2026-03-19T21:37:25.239250Z",
-     "iopub.status.idle": "2026-03-19T21:37:25.392599Z",
-     "shell.execute_reply": "2026-03-19T21:37:25.392251Z"
+     "iopub.execute_input": "2026-03-20T00:20:04.605663Z",
+     "iopub.status.busy": "2026-03-20T00:20:04.605539Z",
+     "iopub.status.idle": "2026-03-20T00:20:04.738595Z",
+     "shell.execute_reply": "2026-03-20T00:20:04.738219Z"
     }
    },
    "outputs": [],
@@ -83,10 +83,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:25.414936Z",
-     "iopub.status.busy": "2026-03-19T21:37:25.414776Z",
-     "iopub.status.idle": "2026-03-19T21:37:26.511943Z",
-     "shell.execute_reply": "2026-03-19T21:37:26.511545Z"
+     "iopub.execute_input": "2026-03-20T00:20:04.758670Z",
+     "iopub.status.busy": "2026-03-20T00:20:04.758387Z",
+     "iopub.status.idle": "2026-03-20T00:20:06.048310Z",
+     "shell.execute_reply": "2026-03-20T00:20:06.047996Z"
     }
    },
    "outputs": [],
@@ -106,10 +106,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:26.513886Z",
-     "iopub.status.busy": "2026-03-19T21:37:26.513710Z",
-     "iopub.status.idle": "2026-03-19T21:37:26.570414Z",
-     "shell.execute_reply": "2026-03-19T21:37:26.570053Z"
+     "iopub.execute_input": "2026-03-20T00:20:06.050717Z",
+     "iopub.status.busy": "2026-03-20T00:20:06.050499Z",
+     "iopub.status.idle": "2026-03-20T00:20:06.107163Z",
+     "shell.execute_reply": "2026-03-20T00:20:06.106879Z"
     }
    },
    "outputs": [],
@@ -143,10 +143,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:26.572173Z",
-     "iopub.status.busy": "2026-03-19T21:37:26.572070Z",
-     "iopub.status.idle": "2026-03-19T21:37:26.610786Z",
-     "shell.execute_reply": "2026-03-19T21:37:26.610502Z"
+     "iopub.execute_input": "2026-03-20T00:20:06.109079Z",
+     "iopub.status.busy": "2026-03-20T00:20:06.108969Z",
+     "iopub.status.idle": "2026-03-20T00:20:06.148206Z",
+     "shell.execute_reply": "2026-03-20T00:20:06.147922Z"
     }
    },
    "outputs": [],
@@ -182,10 +182,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:26.612190Z",
-     "iopub.status.busy": "2026-03-19T21:37:26.612070Z",
-     "iopub.status.idle": "2026-03-19T21:37:26.758827Z",
-     "shell.execute_reply": "2026-03-19T21:37:26.758521Z"
+     "iopub.execute_input": "2026-03-20T00:20:06.149761Z",
+     "iopub.status.busy": "2026-03-20T00:20:06.149643Z",
+     "iopub.status.idle": "2026-03-20T00:20:06.260087Z",
+     "shell.execute_reply": "2026-03-20T00:20:06.259768Z"
     }
    },
    "outputs": [],
@@ -215,10 +215,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:26.760447Z",
-     "iopub.status.busy": "2026-03-19T21:37:26.760316Z",
-     "iopub.status.idle": "2026-03-19T21:37:26.843066Z",
-     "shell.execute_reply": "2026-03-19T21:37:26.842792Z"
+     "iopub.execute_input": "2026-03-20T00:20:06.261616Z",
+     "iopub.status.busy": "2026-03-20T00:20:06.261500Z",
+     "iopub.status.idle": "2026-03-20T00:20:06.347073Z",
+     "shell.execute_reply": "2026-03-20T00:20:06.346703Z"
     }
    },
    "outputs": [],
diff --git a/nbs/data_patch_sequential.ipynb b/nbs/data_patch_sequential.ipynb
index 86f2103..21ac603 100644
--- a/nbs/data_patch_sequential.ipynb
+++ b/nbs/data_patch_sequential.ipynb
@@ -51,10 +51,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:29.000159Z",
-     "iopub.status.busy": "2026-03-19T21:37:28.999233Z",
-     "iopub.status.idle": "2026-03-19T21:37:29.152492Z",
-     "shell.execute_reply": "2026-03-19T21:37:29.152116Z"
+     "iopub.execute_input": "2026-03-20T00:20:08.707656Z",
+     "iopub.status.busy": "2026-03-20T00:20:08.707195Z",
+     "iopub.status.idle": "2026-03-20T00:20:08.871827Z",
+     "shell.execute_reply": "2026-03-20T00:20:08.871466Z"
     }
    },
    "outputs": [],
@@ -84,10 +84,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:29.175831Z",
-     "iopub.status.busy": "2026-03-19T21:37:29.175652Z",
-     "iopub.status.idle": "2026-03-19T21:37:30.266703Z",
-     "shell.execute_reply": "2026-03-19T21:37:30.266342Z"
+     "iopub.execute_input": "2026-03-20T00:20:08.895839Z",
+     "iopub.status.busy": "2026-03-20T00:20:08.895673Z",
+     "iopub.status.idle": "2026-03-20T00:20:09.967789Z",
+     "shell.execute_reply": "2026-03-20T00:20:09.967461Z"
     }
    },
    "outputs": [],
@@ -121,10 +121,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:30.268577Z",
-     "iopub.status.busy": "2026-03-19T21:37:30.268394Z",
-     "iopub.status.idle": "2026-03-19T21:37:30.376158Z",
-     "shell.execute_reply": "2026-03-19T21:37:30.375847Z"
+     "iopub.execute_input": "2026-03-20T00:20:09.970251Z",
+     "iopub.status.busy": "2026-03-20T00:20:09.970017Z",
+     "iopub.status.idle": "2026-03-20T00:20:10.078111Z",
+     "shell.execute_reply": "2026-03-20T00:20:10.077812Z"
     }
    },
    "outputs": [],
@@ -183,10 +183,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:30.379787Z",
-     "iopub.status.busy": "2026-03-19T21:37:30.379669Z",
-     "iopub.status.idle": "2026-03-19T21:37:30.480794Z",
-     "shell.execute_reply": "2026-03-19T21:37:30.480353Z"
+     "iopub.execute_input": "2026-03-20T00:20:10.082184Z",
+     "iopub.status.busy": "2026-03-20T00:20:10.082016Z",
+     "iopub.status.idle": "2026-03-20T00:20:10.188166Z",
+     "shell.execute_reply": "2026-03-20T00:20:10.187891Z"
     }
    },
    "outputs": [],
diff --git a/nbs/descriptors_matching.ipynb b/nbs/descriptors_matching.ipynb
index 6651dfb..ba78ec3 100644
--- a/nbs/descriptors_matching.ipynb
+++ b/nbs/descriptors_matching.ipynb
@@ -63,10 +63,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:32.582902Z",
-     "iopub.status.busy": "2026-03-19T21:37:32.582741Z",
-     "iopub.status.idle": "2026-03-19T21:37:34.180299Z",
-     "shell.execute_reply": "2026-03-19T21:37:34.179862Z"
+     "iopub.execute_input": "2026-03-20T00:20:12.361988Z",
+     "iopub.status.busy": "2026-03-20T00:20:12.361836Z",
+     "iopub.status.idle": "2026-03-20T00:20:13.599122Z",
+     "shell.execute_reply": "2026-03-20T00:20:13.595188Z"
     }
    },
    "outputs": [],
@@ -112,10 +112,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:34.205201Z",
-     "iopub.status.busy": "2026-03-19T21:37:34.205006Z",
-     "iopub.status.idle": "2026-03-19T21:37:35.242039Z",
-     "shell.execute_reply": "2026-03-19T21:37:35.241751Z"
+     "iopub.execute_input": "2026-03-20T00:20:13.629252Z",
+     "iopub.status.busy": "2026-03-20T00:20:13.629063Z",
+     "iopub.status.idle": "2026-03-20T00:20:14.711202Z",
+     "shell.execute_reply": "2026-03-20T00:20:14.710905Z"
     }
    },
    "outputs": [],
@@ -150,10 +150,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:35.243756Z",
-     "iopub.status.busy": "2026-03-19T21:37:35.243621Z",
-     "iopub.status.idle": "2026-03-19T21:37:35.247783Z",
-     "shell.execute_reply": "2026-03-19T21:37:35.247523Z"
+     "iopub.execute_input": "2026-03-20T00:20:14.713070Z",
+     "iopub.status.busy": "2026-03-20T00:20:14.712913Z",
+     "iopub.status.idle": "2026-03-20T00:20:14.717387Z",
+     "shell.execute_reply": "2026-03-20T00:20:14.717092Z"
     }
    },
    "outputs": [],
@@ -210,10 +210,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:35.249085Z",
-     "iopub.status.busy": "2026-03-19T21:37:35.248999Z",
-     "iopub.status.idle": "2026-03-19T21:37:35.697719Z",
-     "shell.execute_reply": "2026-03-19T21:37:35.697265Z"
+     "iopub.execute_input": "2026-03-20T00:20:14.719039Z",
+     "iopub.status.busy": "2026-03-20T00:20:14.718942Z",
+     "iopub.status.idle": "2026-03-20T00:20:15.170737Z",
+     "shell.execute_reply": "2026-03-20T00:20:15.170320Z"
     }
    },
    "outputs": [],
@@ -242,10 +242,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:35.704956Z",
-     "iopub.status.busy": "2026-03-19T21:37:35.704829Z",
-     "iopub.status.idle": "2026-03-19T21:37:35.708745Z",
-     "shell.execute_reply": "2026-03-19T21:37:35.708482Z"
+     "iopub.execute_input": "2026-03-20T00:20:15.177803Z",
+     "iopub.status.busy": "2026-03-20T00:20:15.177670Z",
+     "iopub.status.idle": "2026-03-20T00:20:15.182382Z",
+     "shell.execute_reply": "2026-03-20T00:20:15.181880Z"
     }
    },
    "outputs": [],
@@ -300,10 +300,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:35.710151Z",
-     "iopub.status.busy": "2026-03-19T21:37:35.710058Z",
-     "iopub.status.idle": "2026-03-19T21:37:39.788062Z",
-     "shell.execute_reply": "2026-03-19T21:37:39.787558Z"
+     "iopub.execute_input": "2026-03-20T00:20:15.183910Z",
+     "iopub.status.busy": "2026-03-20T00:20:15.183802Z",
+     "iopub.status.idle": "2026-03-20T00:20:19.766513Z",
+     "shell.execute_reply": "2026-03-20T00:20:19.766007Z"
     }
    },
    "outputs": [],
@@ -325,10 +325,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:39.795473Z",
-     "iopub.status.busy": "2026-03-19T21:37:39.795330Z",
-     "iopub.status.idle": "2026-03-19T21:37:42.451499Z",
-     "shell.execute_reply": "2026-03-19T21:37:42.451133Z"
+     "iopub.execute_input": "2026-03-20T00:20:19.774373Z",
+     "iopub.status.busy": "2026-03-20T00:20:19.774147Z",
+     "iopub.status.idle": "2026-03-20T00:20:21.216762Z",
+     "shell.execute_reply": "2026-03-20T00:20:21.216402Z"
     }
    },
    "outputs": [],
@@ -351,10 +351,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:42.458600Z",
-     "iopub.status.busy": "2026-03-19T21:37:42.458453Z",
-     "iopub.status.idle": "2026-03-19T21:37:50.493439Z",
-     "shell.execute_reply": "2026-03-19T21:37:50.493059Z"
+     "iopub.execute_input": "2026-03-20T00:20:21.223989Z",
+     "iopub.status.busy": "2026-03-20T00:20:21.223842Z",
+     "iopub.status.idle": "2026-03-20T00:20:29.013265Z",
+     "shell.execute_reply": "2026-03-20T00:20:29.012830Z"
     }
    },
    "outputs": [],
@@ -385,10 +385,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:50.500553Z",
-     "iopub.status.busy": "2026-03-19T21:37:50.500423Z",
-     "iopub.status.idle": "2026-03-19T21:37:50.504106Z",
-     "shell.execute_reply": "2026-03-19T21:37:50.503798Z"
+     "iopub.execute_input": "2026-03-20T00:20:29.020883Z",
+     "iopub.status.busy": "2026-03-20T00:20:29.020717Z",
+     "iopub.status.idle": "2026-03-20T00:20:29.024428Z",
+     "shell.execute_reply": "2026-03-20T00:20:29.024140Z"
     }
    },
    "outputs": [],
@@ -431,10 +431,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:37:50.505362Z",
-     "iopub.status.busy": "2026-03-19T21:37:50.505277Z",
-     "iopub.status.idle": "2026-03-19T21:38:01.363537Z",
-     "shell.execute_reply": "2026-03-19T21:38:01.363195Z"
+     "iopub.execute_input": "2026-03-20T00:20:29.025768Z",
+     "iopub.status.busy": "2026-03-20T00:20:29.025673Z",
+     "iopub.status.idle": "2026-03-20T00:20:40.585118Z",
+     "shell.execute_reply": "2026-03-20T00:20:40.584741Z"
     }
    },
    "outputs": [],
@@ -455,10 +455,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:01.370949Z",
-     "iopub.status.busy": "2026-03-19T21:38:01.370796Z",
-     "iopub.status.idle": "2026-03-19T21:38:15.494496Z",
-     "shell.execute_reply": "2026-03-19T21:38:15.494190Z"
+     "iopub.execute_input": "2026-03-20T00:20:40.592759Z",
+     "iopub.status.busy": "2026-03-20T00:20:40.592598Z",
+     "iopub.status.idle": "2026-03-20T00:20:53.508507Z",
+     "shell.execute_reply": "2026-03-20T00:20:53.507980Z"
     }
    },
    "outputs": [],
@@ -478,10 +478,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:15.501770Z",
-     "iopub.status.busy": "2026-03-19T21:38:15.501630Z",
-     "iopub.status.idle": "2026-03-19T21:38:29.119911Z",
-     "shell.execute_reply": "2026-03-19T21:38:29.119608Z"
+     "iopub.execute_input": "2026-03-20T00:20:53.516396Z",
+     "iopub.status.busy": "2026-03-20T00:20:53.516236Z",
+     "iopub.status.idle": "2026-03-20T00:21:05.435105Z",
+     "shell.execute_reply": "2026-03-20T00:21:05.434628Z"
     }
    },
    "outputs": [],
@@ -501,10 +501,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:29.127284Z",
-     "iopub.status.busy": "2026-03-19T21:38:29.127138Z",
-     "iopub.status.idle": "2026-03-19T21:38:51.690508Z",
-     "shell.execute_reply": "2026-03-19T21:38:51.690169Z"
+     "iopub.execute_input": "2026-03-20T00:21:05.443436Z",
+     "iopub.status.busy": "2026-03-20T00:21:05.443249Z",
+     "iopub.status.idle": "2026-03-20T00:21:19.660831Z",
+     "shell.execute_reply": "2026-03-20T00:21:19.660216Z"
     }
    },
    "outputs": [],
diff --git a/nbs/extract_combine_patches.ipynb b/nbs/extract_combine_patches.ipynb
index 2f17004..681e0c8 100644
--- a/nbs/extract_combine_patches.ipynb
+++ b/nbs/extract_combine_patches.ipynb
@@ -53,10 +53,10 @@
    "id": "f02ed2da",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:29.341020Z",
-     "iopub.status.busy": "2026-03-19T21:26:29.340655Z",
-     "iopub.status.idle": "2026-03-19T21:26:29.496453Z",
-     "shell.execute_reply": "2026-03-19T21:26:29.495910Z"
+     "iopub.execute_input": "2026-03-20T00:21:22.226278Z",
+     "iopub.status.busy": "2026-03-20T00:21:22.225952Z",
+     "iopub.status.idle": "2026-03-20T00:21:22.402338Z",
+     "shell.execute_reply": "2026-03-20T00:21:22.401857Z"
     }
    },
    "outputs": [],
@@ -87,10 +87,10 @@
    "id": "033903e6",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:29.498239Z",
-     "iopub.status.busy": "2026-03-19T21:26:29.498085Z",
-     "iopub.status.idle": "2026-03-19T21:26:30.467125Z",
-     "shell.execute_reply": "2026-03-19T21:26:30.466814Z"
+     "iopub.execute_input": "2026-03-20T00:21:22.404127Z",
+     "iopub.status.busy": "2026-03-20T00:21:22.403964Z",
+     "iopub.status.idle": "2026-03-20T00:21:23.746181Z",
+     "shell.execute_reply": "2026-03-20T00:21:23.745500Z"
     }
    },
    "outputs": [],
@@ -123,10 +123,10 @@
    "id": "7419ae3d",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:30.468694Z",
-     "iopub.status.busy": "2026-03-19T21:26:30.468572Z",
-     "iopub.status.idle": "2026-03-19T21:26:30.472784Z",
-     "shell.execute_reply": "2026-03-19T21:26:30.472546Z"
+     "iopub.execute_input": "2026-03-20T00:21:23.748511Z",
+     "iopub.status.busy": "2026-03-20T00:21:23.748309Z",
+     "iopub.status.idle": "2026-03-20T00:21:23.752539Z",
+     "shell.execute_reply": "2026-03-20T00:21:23.752244Z"
     }
    },
    "outputs": [],
@@ -160,10 +160,10 @@
    "id": "8ad0bae6",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:30.474142Z",
-     "iopub.status.busy": "2026-03-19T21:26:30.474020Z",
-     "iopub.status.idle": "2026-03-19T21:26:30.476924Z",
-     "shell.execute_reply": "2026-03-19T21:26:30.476673Z"
+     "iopub.execute_input": "2026-03-20T00:21:23.753877Z",
+     "iopub.status.busy": "2026-03-20T00:21:23.753782Z",
+     "iopub.status.idle": "2026-03-20T00:21:23.757493Z",
+     "shell.execute_reply": "2026-03-20T00:21:23.757194Z"
     }
    },
    "outputs": [],
@@ -216,10 +216,10 @@
    "id": "01add821",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:30.478319Z",
-     "iopub.status.busy": "2026-03-19T21:26:30.478210Z",
-     "iopub.status.idle": "2026-03-19T21:26:30.480403Z",
-     "shell.execute_reply": "2026-03-19T21:26:30.480178Z"
+     "iopub.execute_input": "2026-03-20T00:21:23.759025Z",
+     "iopub.status.busy": "2026-03-20T00:21:23.758912Z",
+     "iopub.status.idle": "2026-03-20T00:21:23.761323Z",
+     "shell.execute_reply": "2026-03-20T00:21:23.761079Z"
     }
    },
    "outputs": [],
@@ -247,10 +247,10 @@
    "id": "52c9c54d",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:30.481572Z",
-     "iopub.status.busy": "2026-03-19T21:26:30.481476Z",
-     "iopub.status.idle": "2026-03-19T21:26:30.483904Z",
-     "shell.execute_reply": "2026-03-19T21:26:30.483657Z"
+     "iopub.execute_input": "2026-03-20T00:21:23.762534Z",
+     "iopub.status.busy": "2026-03-20T00:21:23.762437Z",
+     "iopub.status.idle": "2026-03-20T00:21:23.765148Z",
+     "shell.execute_reply": "2026-03-20T00:21:23.764840Z"
     }
    },
    "outputs": [],
@@ -273,10 +273,10 @@
    "id": "b65659e4",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:30.485148Z",
-     "iopub.status.busy": "2026-03-19T21:26:30.485053Z",
-     "iopub.status.idle": "2026-03-19T21:26:30.486853Z",
-     "shell.execute_reply": "2026-03-19T21:26:30.486606Z"
+     "iopub.execute_input": "2026-03-20T00:21:23.766931Z",
+     "iopub.status.busy": "2026-03-20T00:21:23.766805Z",
+     "iopub.status.idle": "2026-03-20T00:21:23.768878Z",
+     "shell.execute_reply": "2026-03-20T00:21:23.768605Z"
     }
    },
    "outputs": [],
@@ -291,10 +291,10 @@
    "id": "4a07b557",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:30.488011Z",
-     "iopub.status.busy": "2026-03-19T21:26:30.487932Z",
-     "iopub.status.idle": "2026-03-19T21:26:30.490174Z",
-     "shell.execute_reply": "2026-03-19T21:26:30.489949Z"
+     "iopub.execute_input": "2026-03-20T00:21:23.770472Z",
+     "iopub.status.busy": "2026-03-20T00:21:23.770296Z",
+     "iopub.status.idle": "2026-03-20T00:21:23.773256Z",
+     "shell.execute_reply": "2026-03-20T00:21:23.772958Z"
     }
    },
    "outputs": [],
@@ -316,10 +316,10 @@
    "id": "7997ab81",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:30.491512Z",
-     "iopub.status.busy": "2026-03-19T21:26:30.491423Z",
-     "iopub.status.idle": "2026-03-19T21:26:30.493165Z",
-     "shell.execute_reply": "2026-03-19T21:26:30.492939Z"
+     "iopub.execute_input": "2026-03-20T00:21:23.774718Z",
+     "iopub.status.busy": "2026-03-20T00:21:23.774600Z",
+     "iopub.status.idle": "2026-03-20T00:21:23.776503Z",
+     "shell.execute_reply": "2026-03-20T00:21:23.776264Z"
     }
    },
    "outputs": [],
@@ -342,16 +342,16 @@
    "id": "76bf7ecd",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:30.494454Z",
-     "iopub.status.busy": "2026-03-19T21:26:30.494371Z",
-     "iopub.status.idle": "2026-03-19T21:26:30.541417Z",
-     "shell.execute_reply": "2026-03-19T21:26:30.541106Z"
+     "iopub.execute_input": "2026-03-20T00:21:23.777749Z",
+     "iopub.status.busy": "2026-03-20T00:21:23.777635Z",
+     "iopub.status.idle": "2026-03-20T00:21:23.835495Z",
+     "shell.execute_reply": "2026-03-20T00:21:23.835174Z"
     }
    },
    "outputs": [],
    "source": [
     "# Load sample image\n",
-    "img_tensor = K.image_to_tensor(np.array(Image.open(\"panda.jpg\").convert(\"RGB\")))[None, ...]  # BxCxHxW\n",
+    "img_tensor = K.image_to_tensor(np.array(Image.open(\"panda.jpg\").convert(\"RGB\"))).float()[None, ...] / 255.0  # BxCxHxW\n",
     "h, w = img_tensor.shape[-2:]\n",
     "print(f\"Shape of image = {img_tensor.shape}\")\n",
     "\n",
@@ -374,10 +374,10 @@
    "id": "b8a0b022",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:30.542867Z",
-     "iopub.status.busy": "2026-03-19T21:26:30.542779Z",
-     "iopub.status.idle": "2026-03-19T21:26:30.546978Z",
-     "shell.execute_reply": "2026-03-19T21:26:30.546718Z"
+     "iopub.execute_input": "2026-03-20T00:21:23.836906Z",
+     "iopub.status.busy": "2026-03-20T00:21:23.836806Z",
+     "iopub.status.idle": "2026-03-20T00:21:23.845851Z",
+     "shell.execute_reply": "2026-03-20T00:21:23.845533Z"
     }
    },
    "outputs": [],
@@ -400,10 +400,10 @@
    "id": "61eb1842",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:30.548275Z",
-     "iopub.status.busy": "2026-03-19T21:26:30.548188Z",
-     "iopub.status.idle": "2026-03-19T21:26:30.720998Z",
-     "shell.execute_reply": "2026-03-19T21:26:30.720593Z"
+     "iopub.execute_input": "2026-03-20T00:21:23.847525Z",
+     "iopub.status.busy": "2026-03-20T00:21:23.847384Z",
+     "iopub.status.idle": "2026-03-20T00:21:24.002147Z",
+     "shell.execute_reply": "2026-03-20T00:21:24.001737Z"
     }
    },
    "outputs": [],
@@ -433,10 +433,10 @@
    "id": "47e520c8",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:30.722911Z",
-     "iopub.status.busy": "2026-03-19T21:26:30.722749Z",
-     "iopub.status.idle": "2026-03-19T21:26:30.797965Z",
-     "shell.execute_reply": "2026-03-19T21:26:30.797611Z"
+     "iopub.execute_input": "2026-03-20T00:21:24.004338Z",
+     "iopub.status.busy": "2026-03-20T00:21:24.004153Z",
+     "iopub.status.idle": "2026-03-20T00:21:24.075279Z",
+     "shell.execute_reply": "2026-03-20T00:21:24.074925Z"
     }
    },
    "outputs": [],
diff --git a/nbs/face_detection.ipynb b/nbs/face_detection.ipynb
index 0384bdd..a28e890 100644
--- a/nbs/face_detection.ipynb
+++ b/nbs/face_detection.ipynb
@@ -42,10 +42,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:32.906087Z",
-     "iopub.status.busy": "2026-03-19T21:26:32.905312Z",
-     "iopub.status.idle": "2026-03-19T21:26:33.804731Z",
-     "shell.execute_reply": "2026-03-19T21:26:33.803828Z"
+     "iopub.execute_input": "2026-03-20T00:21:26.426651Z",
+     "iopub.status.busy": "2026-03-20T00:21:26.426065Z",
+     "iopub.status.idle": "2026-03-20T00:21:26.793105Z",
+     "shell.execute_reply": "2026-03-20T00:21:26.790918Z"
     }
    },
    "outputs": [],
@@ -82,10 +82,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:33.837351Z",
-     "iopub.status.busy": "2026-03-19T21:26:33.837050Z",
-     "iopub.status.idle": "2026-03-19T21:26:34.948590Z",
-     "shell.execute_reply": "2026-03-19T21:26:34.948052Z"
+     "iopub.execute_input": "2026-03-20T00:21:26.829389Z",
+     "iopub.status.busy": "2026-03-20T00:21:26.829188Z",
+     "iopub.status.idle": "2026-03-20T00:21:27.975068Z",
+     "shell.execute_reply": "2026-03-20T00:21:27.974752Z"
     }
    },
    "outputs": [],
@@ -116,10 +116,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:34.950581Z",
-     "iopub.status.busy": "2026-03-19T21:26:34.950385Z",
-     "iopub.status.idle": "2026-03-19T21:26:34.972184Z",
-     "shell.execute_reply": "2026-03-19T21:26:34.971824Z"
+     "iopub.execute_input": "2026-03-20T00:21:27.976977Z",
+     "iopub.status.busy": "2026-03-20T00:21:27.976790Z",
+     "iopub.status.idle": "2026-03-20T00:21:27.998761Z",
+     "shell.execute_reply": "2026-03-20T00:21:27.998261Z"
     }
    },
    "outputs": [],
@@ -134,10 +134,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:34.974582Z",
-     "iopub.status.busy": "2026-03-19T21:26:34.974378Z",
-     "iopub.status.idle": "2026-03-19T21:26:35.072345Z",
-     "shell.execute_reply": "2026-03-19T21:26:35.071983Z"
+     "iopub.execute_input": "2026-03-20T00:21:28.000858Z",
+     "iopub.status.busy": "2026-03-20T00:21:28.000663Z",
+     "iopub.status.idle": "2026-03-20T00:21:28.098751Z",
+     "shell.execute_reply": "2026-03-20T00:21:28.098247Z"
     }
    },
    "outputs": [],
@@ -160,10 +160,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:35.076836Z",
-     "iopub.status.busy": "2026-03-19T21:26:35.076655Z",
-     "iopub.status.idle": "2026-03-19T21:26:35.261935Z",
-     "shell.execute_reply": "2026-03-19T21:26:35.261557Z"
+     "iopub.execute_input": "2026-03-20T00:21:28.103466Z",
+     "iopub.status.busy": "2026-03-20T00:21:28.103205Z",
+     "iopub.status.idle": "2026-03-20T00:21:28.374131Z",
+     "shell.execute_reply": "2026-03-20T00:21:28.373782Z"
     }
    },
    "outputs": [],
@@ -192,10 +192,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:35.263822Z",
-     "iopub.status.busy": "2026-03-19T21:26:35.263666Z",
-     "iopub.status.idle": "2026-03-19T21:26:35.266144Z",
-     "shell.execute_reply": "2026-03-19T21:26:35.265887Z"
+     "iopub.execute_input": "2026-03-20T00:21:28.375810Z",
+     "iopub.status.busy": "2026-03-20T00:21:28.375700Z",
+     "iopub.status.idle": "2026-03-20T00:21:28.378315Z",
+     "shell.execute_reply": "2026-03-20T00:21:28.378040Z"
     }
    },
    "outputs": [],
@@ -226,10 +226,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:35.267459Z",
-     "iopub.status.busy": "2026-03-19T21:26:35.267347Z",
-     "iopub.status.idle": "2026-03-19T21:26:35.484652Z",
-     "shell.execute_reply": "2026-03-19T21:26:35.484343Z"
+     "iopub.execute_input": "2026-03-20T00:21:28.379770Z",
+     "iopub.status.busy": "2026-03-20T00:21:28.379649Z",
+     "iopub.status.idle": "2026-03-20T00:21:28.594011Z",
+     "shell.execute_reply": "2026-03-20T00:21:28.593675Z"
     }
    },
    "outputs": [],
@@ -273,10 +273,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:35.488484Z",
-     "iopub.status.busy": "2026-03-19T21:26:35.488357Z",
-     "iopub.status.idle": "2026-03-19T21:26:35.818874Z",
-     "shell.execute_reply": "2026-03-19T21:26:35.817890Z"
+     "iopub.execute_input": "2026-03-20T00:21:28.597850Z",
+     "iopub.status.busy": "2026-03-20T00:21:28.597726Z",
+     "iopub.status.idle": "2026-03-20T00:21:28.708873Z",
+     "shell.execute_reply": "2026-03-20T00:21:28.708523Z"
     }
    },
    "outputs": [],
diff --git a/nbs/filtering_edges.ipynb b/nbs/filtering_edges.ipynb
index cf23bc5..99ca388 100644
--- a/nbs/filtering_edges.ipynb
+++ b/nbs/filtering_edges.ipynb
@@ -43,10 +43,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:56.404012Z",
-     "iopub.status.busy": "2026-03-19T21:38:56.395434Z",
-     "iopub.status.idle": "2026-03-19T21:38:57.080048Z",
-     "shell.execute_reply": "2026-03-19T21:38:57.078474Z"
+     "iopub.execute_input": "2026-03-20T00:21:30.878568Z",
+     "iopub.status.busy": "2026-03-20T00:21:30.877162Z",
+     "iopub.status.idle": "2026-03-20T00:21:31.424051Z",
+     "shell.execute_reply": "2026-03-20T00:21:31.423237Z"
     }
    },
    "outputs": [],
@@ -76,10 +76,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:57.100931Z",
-     "iopub.status.busy": "2026-03-19T21:38:57.100713Z",
-     "iopub.status.idle": "2026-03-19T21:38:59.358373Z",
-     "shell.execute_reply": "2026-03-19T21:38:59.358026Z"
+     "iopub.execute_input": "2026-03-20T00:21:31.446329Z",
+     "iopub.status.busy": "2026-03-20T00:21:31.446129Z",
+     "iopub.status.idle": "2026-03-20T00:21:33.307539Z",
+     "shell.execute_reply": "2026-03-20T00:21:33.307056Z"
     }
    },
    "outputs": [],
@@ -107,10 +107,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:59.360023Z",
-     "iopub.status.busy": "2026-03-19T21:38:59.359883Z",
-     "iopub.status.idle": "2026-03-19T21:38:59.368010Z",
-     "shell.execute_reply": "2026-03-19T21:38:59.367694Z"
+     "iopub.execute_input": "2026-03-20T00:21:33.309736Z",
+     "iopub.status.busy": "2026-03-20T00:21:33.309510Z",
+     "iopub.status.idle": "2026-03-20T00:21:33.317206Z",
+     "shell.execute_reply": "2026-03-20T00:21:33.316712Z"
     }
    },
    "outputs": [],
@@ -127,10 +127,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:59.369528Z",
-     "iopub.status.busy": "2026-03-19T21:38:59.369411Z",
-     "iopub.status.idle": "2026-03-19T21:38:59.371442Z",
-     "shell.execute_reply": "2026-03-19T21:38:59.371216Z"
+     "iopub.execute_input": "2026-03-20T00:21:33.319000Z",
+     "iopub.status.busy": "2026-03-20T00:21:33.318864Z",
+     "iopub.status.idle": "2026-03-20T00:21:33.321089Z",
+     "shell.execute_reply": "2026-03-20T00:21:33.320750Z"
     }
    },
    "outputs": [],
@@ -148,10 +148,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:59.372614Z",
-     "iopub.status.busy": "2026-03-19T21:38:59.372521Z",
-     "iopub.status.idle": "2026-03-19T21:38:59.398065Z",
-     "shell.execute_reply": "2026-03-19T21:38:59.397684Z"
+     "iopub.execute_input": "2026-03-20T00:21:33.323136Z",
+     "iopub.status.busy": "2026-03-20T00:21:33.322960Z",
+     "iopub.status.idle": "2026-03-20T00:21:33.348781Z",
+     "shell.execute_reply": "2026-03-20T00:21:33.348497Z"
     }
    },
    "outputs": [],
@@ -171,10 +171,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:59.399771Z",
-     "iopub.status.busy": "2026-03-19T21:38:59.399553Z",
-     "iopub.status.idle": "2026-03-19T21:38:59.402740Z",
-     "shell.execute_reply": "2026-03-19T21:38:59.402399Z"
+     "iopub.execute_input": "2026-03-20T00:21:33.350321Z",
+     "iopub.status.busy": "2026-03-20T00:21:33.350208Z",
+     "iopub.status.idle": "2026-03-20T00:21:33.352942Z",
+     "shell.execute_reply": "2026-03-20T00:21:33.352604Z"
     }
    },
    "outputs": [],
@@ -189,10 +189,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:59.404218Z",
-     "iopub.status.busy": "2026-03-19T21:38:59.404111Z",
-     "iopub.status.idle": "2026-03-19T21:38:59.423660Z",
-     "shell.execute_reply": "2026-03-19T21:38:59.423205Z"
+     "iopub.execute_input": "2026-03-20T00:21:33.355051Z",
+     "iopub.status.busy": "2026-03-20T00:21:33.354843Z",
+     "iopub.status.idle": "2026-03-20T00:21:33.377356Z",
+     "shell.execute_reply": "2026-03-20T00:21:33.376957Z"
     }
    },
    "outputs": [],
@@ -206,10 +206,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:59.425321Z",
-     "iopub.status.busy": "2026-03-19T21:38:59.425189Z",
-     "iopub.status.idle": "2026-03-19T21:38:59.444979Z",
-     "shell.execute_reply": "2026-03-19T21:38:59.444695Z"
+     "iopub.execute_input": "2026-03-20T00:21:33.379016Z",
+     "iopub.status.busy": "2026-03-20T00:21:33.378802Z",
+     "iopub.status.idle": "2026-03-20T00:21:33.399903Z",
+     "shell.execute_reply": "2026-03-20T00:21:33.399384Z"
     }
    },
    "outputs": [],
@@ -230,10 +230,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:59.446691Z",
-     "iopub.status.busy": "2026-03-19T21:38:59.446551Z",
-     "iopub.status.idle": "2026-03-19T21:38:59.449893Z",
-     "shell.execute_reply": "2026-03-19T21:38:59.449561Z"
+     "iopub.execute_input": "2026-03-20T00:21:33.402421Z",
+     "iopub.status.busy": "2026-03-20T00:21:33.402256Z",
+     "iopub.status.idle": "2026-03-20T00:21:33.406686Z",
+     "shell.execute_reply": "2026-03-20T00:21:33.406228Z"
     }
    },
    "outputs": [],
@@ -248,10 +248,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:59.451321Z",
-     "iopub.status.busy": "2026-03-19T21:38:59.451227Z",
-     "iopub.status.idle": "2026-03-19T21:38:59.473068Z",
-     "shell.execute_reply": "2026-03-19T21:38:59.472789Z"
+     "iopub.execute_input": "2026-03-20T00:21:33.408670Z",
+     "iopub.status.busy": "2026-03-20T00:21:33.408520Z",
+     "iopub.status.idle": "2026-03-20T00:21:33.430114Z",
+     "shell.execute_reply": "2026-03-20T00:21:33.429589Z"
     }
    },
    "outputs": [],
@@ -265,10 +265,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:59.474486Z",
-     "iopub.status.busy": "2026-03-19T21:38:59.474359Z",
-     "iopub.status.idle": "2026-03-19T21:38:59.494644Z",
-     "shell.execute_reply": "2026-03-19T21:38:59.494324Z"
+     "iopub.execute_input": "2026-03-20T00:21:33.431724Z",
+     "iopub.status.busy": "2026-03-20T00:21:33.431589Z",
+     "iopub.status.idle": "2026-03-20T00:21:33.452368Z",
+     "shell.execute_reply": "2026-03-20T00:21:33.452028Z"
     }
    },
    "outputs": [],
@@ -290,10 +290,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:59.496215Z",
-     "iopub.status.busy": "2026-03-19T21:38:59.496088Z",
-     "iopub.status.idle": "2026-03-19T21:38:59.516764Z",
-     "shell.execute_reply": "2026-03-19T21:38:59.516434Z"
+     "iopub.execute_input": "2026-03-20T00:21:33.454227Z",
+     "iopub.status.busy": "2026-03-20T00:21:33.454074Z",
+     "iopub.status.idle": "2026-03-20T00:21:33.476113Z",
+     "shell.execute_reply": "2026-03-20T00:21:33.475755Z"
     }
    },
    "outputs": [],
@@ -314,10 +314,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:59.518482Z",
-     "iopub.status.busy": "2026-03-19T21:38:59.518354Z",
-     "iopub.status.idle": "2026-03-19T21:38:59.538694Z",
-     "shell.execute_reply": "2026-03-19T21:38:59.538395Z"
+     "iopub.execute_input": "2026-03-20T00:21:33.477824Z",
+     "iopub.status.busy": "2026-03-20T00:21:33.477654Z",
+     "iopub.status.idle": "2026-03-20T00:21:33.499448Z",
+     "shell.execute_reply": "2026-03-20T00:21:33.499003Z"
     }
    },
    "outputs": [],
@@ -338,10 +338,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:38:59.540289Z",
-     "iopub.status.busy": "2026-03-19T21:38:59.540173Z",
-     "iopub.status.idle": "2026-03-19T21:38:59.574532Z",
-     "shell.execute_reply": "2026-03-19T21:38:59.574206Z"
+     "iopub.execute_input": "2026-03-20T00:21:33.501974Z",
+     "iopub.status.busy": "2026-03-20T00:21:33.501671Z",
+     "iopub.status.idle": "2026-03-20T00:21:33.539392Z",
+     "shell.execute_reply": "2026-03-20T00:21:33.538932Z"
     }
    },
    "outputs": [],
diff --git a/nbs/filtering_operators.ipynb b/nbs/filtering_operators.ipynb
index 33f6b7f..c0e9cc3 100644
--- a/nbs/filtering_operators.ipynb
+++ b/nbs/filtering_operators.ipynb
@@ -44,10 +44,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:01.974208Z",
-     "iopub.status.busy": "2026-03-19T21:39:01.974034Z",
-     "iopub.status.idle": "2026-03-19T21:39:02.240602Z",
-     "shell.execute_reply": "2026-03-19T21:39:02.240286Z"
+     "iopub.execute_input": "2026-03-20T00:21:35.995519Z",
+     "iopub.status.busy": "2026-03-20T00:21:35.994480Z",
+     "iopub.status.idle": "2026-03-20T00:21:36.285043Z",
+     "shell.execute_reply": "2026-03-20T00:21:36.284691Z"
     }
    },
    "outputs": [],
@@ -77,10 +77,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:02.259270Z",
-     "iopub.status.busy": "2026-03-19T21:39:02.259110Z",
-     "iopub.status.idle": "2026-03-19T21:39:03.822034Z",
-     "shell.execute_reply": "2026-03-19T21:39:03.821721Z"
+     "iopub.execute_input": "2026-03-20T00:21:36.310351Z",
+     "iopub.status.busy": "2026-03-20T00:21:36.310172Z",
+     "iopub.status.idle": "2026-03-20T00:21:37.907207Z",
+     "shell.execute_reply": "2026-03-20T00:21:37.906860Z"
     }
    },
    "outputs": [],
@@ -107,10 +107,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:03.823743Z",
-     "iopub.status.busy": "2026-03-19T21:39:03.823567Z",
-     "iopub.status.idle": "2026-03-19T21:39:03.830186Z",
-     "shell.execute_reply": "2026-03-19T21:39:03.829792Z"
+     "iopub.execute_input": "2026-03-20T00:21:37.909260Z",
+     "iopub.status.busy": "2026-03-20T00:21:37.909048Z",
+     "iopub.status.idle": "2026-03-20T00:21:37.915920Z",
+     "shell.execute_reply": "2026-03-20T00:21:37.915455Z"
     }
    },
    "outputs": [],
@@ -127,10 +127,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:03.831825Z",
-     "iopub.status.busy": "2026-03-19T21:39:03.831726Z",
-     "iopub.status.idle": "2026-03-19T21:39:03.834094Z",
-     "shell.execute_reply": "2026-03-19T21:39:03.833859Z"
+     "iopub.execute_input": "2026-03-20T00:21:37.918313Z",
+     "iopub.status.busy": "2026-03-20T00:21:37.918119Z",
+     "iopub.status.idle": "2026-03-20T00:21:37.920946Z",
+     "shell.execute_reply": "2026-03-20T00:21:37.920537Z"
     }
    },
    "outputs": [],
@@ -151,10 +151,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:03.835363Z",
-     "iopub.status.busy": "2026-03-19T21:39:03.835260Z",
-     "iopub.status.idle": "2026-03-19T21:39:03.862948Z",
-     "shell.execute_reply": "2026-03-19T21:39:03.862647Z"
+     "iopub.execute_input": "2026-03-20T00:21:37.922722Z",
+     "iopub.status.busy": "2026-03-20T00:21:37.922592Z",
+     "iopub.status.idle": "2026-03-20T00:21:37.949648Z",
+     "shell.execute_reply": "2026-03-20T00:21:37.949218Z"
     }
    },
    "outputs": [],
@@ -174,10 +174,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:03.864519Z",
-     "iopub.status.busy": "2026-03-19T21:39:03.864362Z",
-     "iopub.status.idle": "2026-03-19T21:39:03.891678Z",
-     "shell.execute_reply": "2026-03-19T21:39:03.891397Z"
+     "iopub.execute_input": "2026-03-20T00:21:37.951318Z",
+     "iopub.status.busy": "2026-03-20T00:21:37.951207Z",
+     "iopub.status.idle": "2026-03-20T00:21:37.979361Z",
+     "shell.execute_reply": "2026-03-20T00:21:37.979046Z"
     }
    },
    "outputs": [],
@@ -198,10 +198,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:03.893090Z",
-     "iopub.status.busy": "2026-03-19T21:39:03.892970Z",
-     "iopub.status.idle": "2026-03-19T21:39:03.918579Z",
-     "shell.execute_reply": "2026-03-19T21:39:03.918310Z"
+     "iopub.execute_input": "2026-03-20T00:21:37.980863Z",
+     "iopub.status.busy": "2026-03-20T00:21:37.980746Z",
+     "iopub.status.idle": "2026-03-20T00:21:38.008362Z",
+     "shell.execute_reply": "2026-03-20T00:21:38.008094Z"
     }
    },
    "outputs": [],
@@ -222,10 +222,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:03.920181Z",
-     "iopub.status.busy": "2026-03-19T21:39:03.920044Z",
-     "iopub.status.idle": "2026-03-19T21:39:03.945611Z",
-     "shell.execute_reply": "2026-03-19T21:39:03.945321Z"
+     "iopub.execute_input": "2026-03-20T00:21:38.010126Z",
+     "iopub.status.busy": "2026-03-20T00:21:38.010009Z",
+     "iopub.status.idle": "2026-03-20T00:21:38.036099Z",
+     "shell.execute_reply": "2026-03-20T00:21:38.035791Z"
     }
    },
    "outputs": [],
@@ -246,10 +246,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:03.947064Z",
-     "iopub.status.busy": "2026-03-19T21:39:03.946886Z",
-     "iopub.status.idle": "2026-03-19T21:39:03.974197Z",
-     "shell.execute_reply": "2026-03-19T21:39:03.973916Z"
+     "iopub.execute_input": "2026-03-20T00:21:38.037647Z",
+     "iopub.status.busy": "2026-03-20T00:21:38.037531Z",
+     "iopub.status.idle": "2026-03-20T00:21:38.065248Z",
+     "shell.execute_reply": "2026-03-20T00:21:38.064956Z"
     }
    },
    "outputs": [],
@@ -270,10 +270,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:03.975595Z",
-     "iopub.status.busy": "2026-03-19T21:39:03.975477Z",
-     "iopub.status.idle": "2026-03-19T21:39:04.002979Z",
-     "shell.execute_reply": "2026-03-19T21:39:04.002667Z"
+     "iopub.execute_input": "2026-03-20T00:21:38.066897Z",
+     "iopub.status.busy": "2026-03-20T00:21:38.066761Z",
+     "iopub.status.idle": "2026-03-20T00:21:38.094118Z",
+     "shell.execute_reply": "2026-03-20T00:21:38.093820Z"
     }
    },
    "outputs": [],
@@ -294,10 +294,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:04.004456Z",
-     "iopub.status.busy": "2026-03-19T21:39:04.004337Z",
-     "iopub.status.idle": "2026-03-19T21:39:04.032812Z",
-     "shell.execute_reply": "2026-03-19T21:39:04.032529Z"
+     "iopub.execute_input": "2026-03-20T00:21:38.095505Z",
+     "iopub.status.busy": "2026-03-20T00:21:38.095398Z",
+     "iopub.status.idle": "2026-03-20T00:21:38.123225Z",
+     "shell.execute_reply": "2026-03-20T00:21:38.122916Z"
     }
    },
    "outputs": [],
diff --git a/nbs/fit_line.ipynb b/nbs/fit_line.ipynb
index 3f5b857..936ee81 100644
--- a/nbs/fit_line.ipynb
+++ b/nbs/fit_line.ipynb
@@ -26,10 +26,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:48.197693Z",
-     "iopub.status.busy": "2026-03-19T21:26:48.197479Z",
-     "iopub.status.idle": "2026-03-19T21:26:49.254382Z",
-     "shell.execute_reply": "2026-03-19T21:26:49.254084Z"
+     "iopub.execute_input": "2026-03-20T00:21:40.461718Z",
+     "iopub.status.busy": "2026-03-20T00:21:40.461321Z",
+     "iopub.status.idle": "2026-03-20T00:21:41.522988Z",
+     "shell.execute_reply": "2026-03-20T00:21:41.522666Z"
     }
    },
    "outputs": [],
@@ -44,10 +44,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:49.256036Z",
-     "iopub.status.busy": "2026-03-19T21:26:49.255915Z",
-     "iopub.status.idle": "2026-03-19T21:26:49.257747Z",
-     "shell.execute_reply": "2026-03-19T21:26:49.257507Z"
+     "iopub.execute_input": "2026-03-20T00:21:41.524684Z",
+     "iopub.status.busy": "2026-03-20T00:21:41.524541Z",
+     "iopub.status.idle": "2026-03-20T00:21:41.526454Z",
+     "shell.execute_reply": "2026-03-20T00:21:41.526191Z"
     }
    },
    "outputs": [],
@@ -61,10 +61,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:49.258999Z",
-     "iopub.status.busy": "2026-03-19T21:26:49.258921Z",
-     "iopub.status.idle": "2026-03-19T21:26:49.263222Z",
-     "shell.execute_reply": "2026-03-19T21:26:49.262959Z"
+     "iopub.execute_input": "2026-03-20T00:21:41.527743Z",
+     "iopub.status.busy": "2026-03-20T00:21:41.527662Z",
+     "iopub.status.idle": "2026-03-20T00:21:41.531123Z",
+     "shell.execute_reply": "2026-03-20T00:21:41.530858Z"
     }
    },
    "outputs": [],
@@ -82,10 +82,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:49.280383Z",
-     "iopub.status.busy": "2026-03-19T21:26:49.280248Z",
-     "iopub.status.idle": "2026-03-19T21:26:49.283800Z",
-     "shell.execute_reply": "2026-03-19T21:26:49.283506Z"
+     "iopub.execute_input": "2026-03-20T00:21:41.548460Z",
+     "iopub.status.busy": "2026-03-20T00:21:41.548312Z",
+     "iopub.status.idle": "2026-03-20T00:21:41.551986Z",
+     "shell.execute_reply": "2026-03-20T00:21:41.551723Z"
     }
    },
    "outputs": [],
@@ -107,10 +107,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:49.285064Z",
-     "iopub.status.busy": "2026-03-19T21:26:49.284952Z",
-     "iopub.status.idle": "2026-03-19T21:26:49.290270Z",
-     "shell.execute_reply": "2026-03-19T21:26:49.290007Z"
+     "iopub.execute_input": "2026-03-20T00:21:41.553263Z",
+     "iopub.status.busy": "2026-03-20T00:21:41.553177Z",
+     "iopub.status.idle": "2026-03-20T00:21:41.556132Z",
+     "shell.execute_reply": "2026-03-20T00:21:41.555894Z"
     }
    },
    "outputs": [],
@@ -129,10 +129,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:49.291432Z",
-     "iopub.status.busy": "2026-03-19T21:26:49.291335Z",
-     "iopub.status.idle": "2026-03-19T21:26:49.335690Z",
-     "shell.execute_reply": "2026-03-19T21:26:49.335417Z"
+     "iopub.execute_input": "2026-03-20T00:21:41.557387Z",
+     "iopub.status.busy": "2026-03-20T00:21:41.557303Z",
+     "iopub.status.idle": "2026-03-20T00:21:41.602642Z",
+     "shell.execute_reply": "2026-03-20T00:21:41.602191Z"
     }
    },
    "outputs": [],
diff --git a/nbs/fit_plane.ipynb b/nbs/fit_plane.ipynb
index 62e4e12..0222063 100644
--- a/nbs/fit_plane.ipynb
+++ b/nbs/fit_plane.ipynb
@@ -26,10 +26,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:51.495810Z",
-     "iopub.status.busy": "2026-03-19T21:26:51.495184Z",
-     "iopub.status.idle": "2026-03-19T21:26:53.147594Z",
-     "shell.execute_reply": "2026-03-19T21:26:53.147136Z"
+     "iopub.execute_input": "2026-03-20T00:21:43.774416Z",
+     "iopub.status.busy": "2026-03-20T00:21:43.774120Z",
+     "iopub.status.idle": "2026-03-20T00:21:45.157784Z",
+     "shell.execute_reply": "2026-03-20T00:21:45.157477Z"
     }
    },
    "outputs": [],
@@ -48,10 +48,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:53.149577Z",
-     "iopub.status.busy": "2026-03-19T21:26:53.149283Z",
-     "iopub.status.idle": "2026-03-19T21:26:53.152361Z",
-     "shell.execute_reply": "2026-03-19T21:26:53.152035Z"
+     "iopub.execute_input": "2026-03-20T00:21:45.159567Z",
+     "iopub.status.busy": "2026-03-20T00:21:45.159343Z",
+     "iopub.status.idle": "2026-03-20T00:21:45.162094Z",
+     "shell.execute_reply": "2026-03-20T00:21:45.161862Z"
     }
    },
    "outputs": [],
@@ -73,10 +73,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:53.154018Z",
-     "iopub.status.busy": "2026-03-19T21:26:53.153894Z",
-     "iopub.status.idle": "2026-03-19T21:26:53.156388Z",
-     "shell.execute_reply": "2026-03-19T21:26:53.156080Z"
+     "iopub.execute_input": "2026-03-20T00:21:45.163392Z",
+     "iopub.status.busy": "2026-03-20T00:21:45.163282Z",
+     "iopub.status.idle": "2026-03-20T00:21:45.165383Z",
+     "shell.execute_reply": "2026-03-20T00:21:45.165136Z"
     }
    },
    "outputs": [],
@@ -93,10 +93,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:53.158220Z",
-     "iopub.status.busy": "2026-03-19T21:26:53.158118Z",
-     "iopub.status.idle": "2026-03-19T21:26:53.642152Z",
-     "shell.execute_reply": "2026-03-19T21:26:53.641645Z"
+     "iopub.execute_input": "2026-03-20T00:21:45.166695Z",
+     "iopub.status.busy": "2026-03-20T00:21:45.166573Z",
+     "iopub.status.idle": "2026-03-20T00:21:45.520765Z",
+     "shell.execute_reply": "2026-03-20T00:21:45.520369Z"
     }
    },
    "outputs": [],
@@ -119,10 +119,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:53.693090Z",
-     "iopub.status.busy": "2026-03-19T21:26:53.692860Z",
-     "iopub.status.idle": "2026-03-19T21:26:53.698201Z",
-     "shell.execute_reply": "2026-03-19T21:26:53.697718Z"
+     "iopub.execute_input": "2026-03-20T00:21:45.571612Z",
+     "iopub.status.busy": "2026-03-20T00:21:45.571450Z",
+     "iopub.status.idle": "2026-03-20T00:21:45.575597Z",
+     "shell.execute_reply": "2026-03-20T00:21:45.575366Z"
     }
    },
    "outputs": [],
@@ -140,10 +140,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:53.700231Z",
-     "iopub.status.busy": "2026-03-19T21:26:53.700065Z",
-     "iopub.status.idle": "2026-03-19T21:26:53.866032Z",
-     "shell.execute_reply": "2026-03-19T21:26:53.865692Z"
+     "iopub.execute_input": "2026-03-20T00:21:45.576850Z",
+     "iopub.status.busy": "2026-03-20T00:21:45.576769Z",
+     "iopub.status.idle": "2026-03-20T00:21:45.769220Z",
+     "shell.execute_reply": "2026-03-20T00:21:45.768895Z"
     }
    },
    "outputs": [],
@@ -159,10 +159,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:53.867644Z",
-     "iopub.status.busy": "2026-03-19T21:26:53.867539Z",
-     "iopub.status.idle": "2026-03-19T21:26:53.897364Z",
-     "shell.execute_reply": "2026-03-19T21:26:53.896928Z"
+     "iopub.execute_input": "2026-03-20T00:21:45.770923Z",
+     "iopub.status.busy": "2026-03-20T00:21:45.770803Z",
+     "iopub.status.idle": "2026-03-20T00:21:45.799744Z",
+     "shell.execute_reply": "2026-03-20T00:21:45.799413Z"
     }
    },
    "outputs": [],
@@ -191,10 +191,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:53.901611Z",
-     "iopub.status.busy": "2026-03-19T21:26:53.901441Z",
-     "iopub.status.idle": "2026-03-19T21:26:53.908753Z",
-     "shell.execute_reply": "2026-03-19T21:26:53.908468Z"
+     "iopub.execute_input": "2026-03-20T00:21:45.804248Z",
+     "iopub.status.busy": "2026-03-20T00:21:45.804102Z",
+     "iopub.status.idle": "2026-03-20T00:21:45.808551Z",
+     "shell.execute_reply": "2026-03-20T00:21:45.808285Z"
     }
    },
    "outputs": [],
@@ -209,10 +209,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:53.910107Z",
-     "iopub.status.busy": "2026-03-19T21:26:53.910005Z",
-     "iopub.status.idle": "2026-03-19T21:26:53.911978Z",
-     "shell.execute_reply": "2026-03-19T21:26:53.911744Z"
+     "iopub.execute_input": "2026-03-20T00:21:45.809778Z",
+     "iopub.status.busy": "2026-03-20T00:21:45.809667Z",
+     "iopub.status.idle": "2026-03-20T00:21:45.811647Z",
+     "shell.execute_reply": "2026-03-20T00:21:45.811418Z"
     }
    },
    "outputs": [],
@@ -226,10 +226,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:53.913426Z",
-     "iopub.status.busy": "2026-03-19T21:26:53.913293Z",
-     "iopub.status.idle": "2026-03-19T21:26:53.945010Z",
-     "shell.execute_reply": "2026-03-19T21:26:53.944560Z"
+     "iopub.execute_input": "2026-03-20T00:21:45.812769Z",
+     "iopub.status.busy": "2026-03-20T00:21:45.812689Z",
+     "iopub.status.idle": "2026-03-20T00:21:45.843699Z",
+     "shell.execute_reply": "2026-03-20T00:21:45.843412Z"
     }
    },
    "outputs": [],
diff --git a/nbs/gaussian_blur.ipynb b/nbs/gaussian_blur.ipynb
index fb49a95..8317501 100644
--- a/nbs/gaussian_blur.ipynb
+++ b/nbs/gaussian_blur.ipynb
@@ -57,10 +57,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:56.167712Z",
-     "iopub.status.busy": "2026-03-19T21:26:56.166945Z",
-     "iopub.status.idle": "2026-03-19T21:26:56.975872Z",
-     "shell.execute_reply": "2026-03-19T21:26:56.975529Z"
+     "iopub.execute_input": "2026-03-20T00:21:48.064527Z",
+     "iopub.status.busy": "2026-03-20T00:21:48.064063Z",
+     "iopub.status.idle": "2026-03-20T00:21:48.958053Z",
+     "shell.execute_reply": "2026-03-20T00:21:48.957592Z"
     }
    },
    "outputs": [],
@@ -84,10 +84,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:56.993511Z",
-     "iopub.status.busy": "2026-03-19T21:26:56.993315Z",
-     "iopub.status.idle": "2026-03-19T21:26:57.697062Z",
-     "shell.execute_reply": "2026-03-19T21:26:57.696151Z"
+     "iopub.execute_input": "2026-03-20T00:21:48.977459Z",
+     "iopub.status.busy": "2026-03-20T00:21:48.977223Z",
+     "iopub.status.idle": "2026-03-20T00:21:49.528129Z",
+     "shell.execute_reply": "2026-03-20T00:21:49.527489Z"
     }
    },
    "outputs": [],
@@ -127,10 +127,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:57.704525Z",
-     "iopub.status.busy": "2026-03-19T21:26:57.704308Z",
-     "iopub.status.idle": "2026-03-19T21:26:57.917702Z",
-     "shell.execute_reply": "2026-03-19T21:26:57.917389Z"
+     "iopub.execute_input": "2026-03-20T00:21:49.531790Z",
+     "iopub.status.busy": "2026-03-20T00:21:49.531362Z",
+     "iopub.status.idle": "2026-03-20T00:21:49.774349Z",
+     "shell.execute_reply": "2026-03-20T00:21:49.773921Z"
     }
    },
    "outputs": [],
@@ -139,7 +139,7 @@
     "import torch\n",
     "\n",
     "# read the image with kornia\n",
-    "data = kornia.image_to_tensor(np.array(Image.open(\"./bennett_aden.png\").convert(\"RGB\")))[None, ...]  # BxCxHxW"
+    "data = kornia.image_to_tensor(np.array(Image.open(\"./bennett_aden.png\").convert(\"RGB\"))).float()[None, ...] / 255.0  # BxCxHxW"
    ]
   },
   {
@@ -154,10 +154,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:57.919334Z",
-     "iopub.status.busy": "2026-03-19T21:26:57.919205Z",
-     "iopub.status.idle": "2026-03-19T21:26:57.923099Z",
-     "shell.execute_reply": "2026-03-19T21:26:57.922824Z"
+     "iopub.execute_input": "2026-03-20T00:21:49.776349Z",
+     "iopub.status.busy": "2026-03-20T00:21:49.776129Z",
+     "iopub.status.idle": "2026-03-20T00:21:49.782740Z",
+     "shell.execute_reply": "2026-03-20T00:21:49.782310Z"
     }
    },
    "outputs": [],
@@ -181,10 +181,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:26:57.924533Z",
-     "iopub.status.busy": "2026-03-19T21:26:57.924448Z",
-     "iopub.status.idle": "2026-03-19T21:26:58.015905Z",
-     "shell.execute_reply": "2026-03-19T21:26:58.015602Z"
+     "iopub.execute_input": "2026-03-20T00:21:49.784794Z",
+     "iopub.status.busy": "2026-03-20T00:21:49.784612Z",
+     "iopub.status.idle": "2026-03-20T00:21:49.901739Z",
+     "shell.execute_reply": "2026-03-20T00:21:49.901274Z"
     }
    },
    "outputs": [],
diff --git a/nbs/geometric_transforms.ipynb b/nbs/geometric_transforms.ipynb
index e69529f..454fe04 100644
--- a/nbs/geometric_transforms.ipynb
+++ b/nbs/geometric_transforms.ipynb
@@ -81,10 +81,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:06.263068Z",
-     "iopub.status.busy": "2026-03-19T21:39:06.262841Z",
-     "iopub.status.idle": "2026-03-19T21:39:12.926726Z",
-     "shell.execute_reply": "2026-03-19T21:39:12.926237Z"
+     "iopub.execute_input": "2026-03-20T00:21:52.039678Z",
+     "iopub.status.busy": "2026-03-20T00:21:52.039084Z",
+     "iopub.status.idle": "2026-03-20T00:21:55.916466Z",
+     "shell.execute_reply": "2026-03-20T00:21:55.915584Z"
     }
    },
    "outputs": [],
@@ -125,10 +125,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:12.953422Z",
-     "iopub.status.busy": "2026-03-19T21:39:12.953240Z",
-     "iopub.status.idle": "2026-03-19T21:39:14.042625Z",
-     "shell.execute_reply": "2026-03-19T21:39:14.042306Z"
+     "iopub.execute_input": "2026-03-20T00:21:55.950587Z",
+     "iopub.status.busy": "2026-03-20T00:21:55.950275Z",
+     "iopub.status.idle": "2026-03-20T00:21:57.088923Z",
+     "shell.execute_reply": "2026-03-20T00:21:57.088507Z"
     }
    },
    "outputs": [],
@@ -157,10 +157,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:14.044642Z",
-     "iopub.status.busy": "2026-03-19T21:39:14.044337Z",
-     "iopub.status.idle": "2026-03-19T21:39:14.046974Z",
-     "shell.execute_reply": "2026-03-19T21:39:14.046585Z"
+     "iopub.execute_input": "2026-03-20T00:21:57.090996Z",
+     "iopub.status.busy": "2026-03-20T00:21:57.090816Z",
+     "iopub.status.idle": "2026-03-20T00:21:57.093356Z",
+     "shell.execute_reply": "2026-03-20T00:21:57.093083Z"
     }
    },
    "outputs": [],
@@ -185,10 +185,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:14.048802Z",
-     "iopub.status.busy": "2026-03-19T21:39:14.048671Z",
-     "iopub.status.idle": "2026-03-19T21:39:14.051459Z",
-     "shell.execute_reply": "2026-03-19T21:39:14.050998Z"
+     "iopub.execute_input": "2026-03-20T00:21:57.094973Z",
+     "iopub.status.busy": "2026-03-20T00:21:57.094844Z",
+     "iopub.status.idle": "2026-03-20T00:21:57.097445Z",
+     "shell.execute_reply": "2026-03-20T00:21:57.097130Z"
     }
    },
    "outputs": [],
@@ -225,10 +225,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:14.053229Z",
-     "iopub.status.busy": "2026-03-19T21:39:14.053096Z",
-     "iopub.status.idle": "2026-03-19T21:39:14.149166Z",
-     "shell.execute_reply": "2026-03-19T21:39:14.148748Z"
+     "iopub.execute_input": "2026-03-20T00:21:57.099077Z",
+     "iopub.status.busy": "2026-03-20T00:21:57.098939Z",
+     "iopub.status.idle": "2026-03-20T00:21:57.194899Z",
+     "shell.execute_reply": "2026-03-20T00:21:57.194602Z"
     }
    },
    "outputs": [],
@@ -262,10 +262,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:14.155262Z",
-     "iopub.status.busy": "2026-03-19T21:39:14.155095Z",
-     "iopub.status.idle": "2026-03-19T21:39:14.325486Z",
-     "shell.execute_reply": "2026-03-19T21:39:14.325053Z"
+     "iopub.execute_input": "2026-03-20T00:21:57.200407Z",
+     "iopub.status.busy": "2026-03-20T00:21:57.200262Z",
+     "iopub.status.idle": "2026-03-20T00:21:57.379074Z",
+     "shell.execute_reply": "2026-03-20T00:21:57.378773Z"
     }
    },
    "outputs": [],
@@ -313,10 +313,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:14.334291Z",
-     "iopub.status.busy": "2026-03-19T21:39:14.334109Z",
-     "iopub.status.idle": "2026-03-19T21:39:14.337357Z",
-     "shell.execute_reply": "2026-03-19T21:39:14.337047Z"
+     "iopub.execute_input": "2026-03-20T00:21:57.389999Z",
+     "iopub.status.busy": "2026-03-20T00:21:57.389782Z",
+     "iopub.status.idle": "2026-03-20T00:21:57.393548Z",
+     "shell.execute_reply": "2026-03-20T00:21:57.393156Z"
     }
    },
    "outputs": [],
@@ -360,10 +360,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:14.338684Z",
-     "iopub.status.busy": "2026-03-19T21:39:14.338600Z",
-     "iopub.status.idle": "2026-03-19T21:39:14.430194Z",
-     "shell.execute_reply": "2026-03-19T21:39:14.429864Z"
+     "iopub.execute_input": "2026-03-20T00:21:57.395164Z",
+     "iopub.status.busy": "2026-03-20T00:21:57.395038Z",
+     "iopub.status.idle": "2026-03-20T00:21:57.493223Z",
+     "shell.execute_reply": "2026-03-20T00:21:57.492882Z"
     }
    },
    "outputs": [],
@@ -391,10 +391,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:14.434663Z",
-     "iopub.status.busy": "2026-03-19T21:39:14.434522Z",
-     "iopub.status.idle": "2026-03-19T21:39:15.093045Z",
-     "shell.execute_reply": "2026-03-19T21:39:15.092743Z"
+     "iopub.execute_input": "2026-03-20T00:21:57.498089Z",
+     "iopub.status.busy": "2026-03-20T00:21:57.497950Z",
+     "iopub.status.idle": "2026-03-20T00:21:58.187510Z",
+     "shell.execute_reply": "2026-03-20T00:21:58.187123Z"
     }
    },
    "outputs": [],
@@ -459,10 +459,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:15.097676Z",
-     "iopub.status.busy": "2026-03-19T21:39:15.097546Z",
-     "iopub.status.idle": "2026-03-19T21:39:15.161313Z",
-     "shell.execute_reply": "2026-03-19T21:39:15.161001Z"
+     "iopub.execute_input": "2026-03-20T00:21:58.191307Z",
+     "iopub.status.busy": "2026-03-20T00:21:58.191182Z",
+     "iopub.status.idle": "2026-03-20T00:21:58.256058Z",
+     "shell.execute_reply": "2026-03-20T00:21:58.255761Z"
     }
    },
    "outputs": [],
@@ -487,10 +487,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:15.164471Z",
-     "iopub.status.busy": "2026-03-19T21:39:15.164351Z",
-     "iopub.status.idle": "2026-03-19T21:39:15.218654Z",
-     "shell.execute_reply": "2026-03-19T21:39:15.218352Z"
+     "iopub.execute_input": "2026-03-20T00:21:58.259191Z",
+     "iopub.status.busy": "2026-03-20T00:21:58.259071Z",
+     "iopub.status.idle": "2026-03-20T00:21:58.317147Z",
+     "shell.execute_reply": "2026-03-20T00:21:58.316804Z"
     }
    },
    "outputs": [],
@@ -519,10 +519,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:15.222038Z",
-     "iopub.status.busy": "2026-03-19T21:39:15.221909Z",
-     "iopub.status.idle": "2026-03-19T21:39:15.285974Z",
-     "shell.execute_reply": "2026-03-19T21:39:15.285613Z"
+     "iopub.execute_input": "2026-03-20T00:21:58.321015Z",
+     "iopub.status.busy": "2026-03-20T00:21:58.320849Z",
+     "iopub.status.idle": "2026-03-20T00:21:58.387134Z",
+     "shell.execute_reply": "2026-03-20T00:21:58.386654Z"
     }
    },
    "outputs": [],
diff --git a/nbs/geometry_generate_patch.ipynb b/nbs/geometry_generate_patch.ipynb
index 206c07d..590ea13 100644
--- a/nbs/geometry_generate_patch.ipynb
+++ b/nbs/geometry_generate_patch.ipynb
@@ -41,10 +41,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:17.424966Z",
-     "iopub.status.busy": "2026-03-19T21:39:17.424816Z",
-     "iopub.status.idle": "2026-03-19T21:39:17.979694Z",
-     "shell.execute_reply": "2026-03-19T21:39:17.979332Z"
+     "iopub.execute_input": "2026-03-20T00:22:00.801566Z",
+     "iopub.status.busy": "2026-03-20T00:22:00.801389Z",
+     "iopub.status.idle": "2026-03-20T00:22:02.119152Z",
+     "shell.execute_reply": "2026-03-20T00:22:02.118712Z"
     }
    },
    "outputs": [],
@@ -80,10 +80,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:18.005587Z",
-     "iopub.status.busy": "2026-03-19T21:39:18.005406Z",
-     "iopub.status.idle": "2026-03-19T21:39:19.036090Z",
-     "shell.execute_reply": "2026-03-19T21:39:19.035777Z"
+     "iopub.execute_input": "2026-03-20T00:22:02.145241Z",
+     "iopub.status.busy": "2026-03-20T00:22:02.145053Z",
+     "iopub.status.idle": "2026-03-20T00:22:03.190152Z",
+     "shell.execute_reply": "2026-03-20T00:22:03.189798Z"
     }
    },
    "outputs": [],
@@ -102,10 +102,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:19.037759Z",
-     "iopub.status.busy": "2026-03-19T21:39:19.037601Z",
-     "iopub.status.idle": "2026-03-19T21:39:19.039797Z",
-     "shell.execute_reply": "2026-03-19T21:39:19.039510Z"
+     "iopub.execute_input": "2026-03-20T00:22:03.191778Z",
+     "iopub.status.busy": "2026-03-20T00:22:03.191640Z",
+     "iopub.status.idle": "2026-03-20T00:22:03.193801Z",
+     "shell.execute_reply": "2026-03-20T00:22:03.193568Z"
     }
    },
    "outputs": [],
@@ -130,10 +130,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:19.041239Z",
-     "iopub.status.busy": "2026-03-19T21:39:19.041134Z",
-     "iopub.status.idle": "2026-03-19T21:39:19.130188Z",
-     "shell.execute_reply": "2026-03-19T21:39:19.129885Z"
+     "iopub.execute_input": "2026-03-20T00:22:03.195104Z",
+     "iopub.status.busy": "2026-03-20T00:22:03.195021Z",
+     "iopub.status.idle": "2026-03-20T00:22:03.286797Z",
+     "shell.execute_reply": "2026-03-20T00:22:03.286452Z"
     }
    },
    "outputs": [],
@@ -157,10 +157,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:19.136048Z",
-     "iopub.status.busy": "2026-03-19T21:39:19.135920Z",
-     "iopub.status.idle": "2026-03-19T21:39:19.185835Z",
-     "shell.execute_reply": "2026-03-19T21:39:19.185482Z"
+     "iopub.execute_input": "2026-03-20T00:22:03.292228Z",
+     "iopub.status.busy": "2026-03-20T00:22:03.292100Z",
+     "iopub.status.idle": "2026-03-20T00:22:03.344767Z",
+     "shell.execute_reply": "2026-03-20T00:22:03.344404Z"
     }
    },
    "outputs": [],
@@ -184,10 +184,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:19.188300Z",
-     "iopub.status.busy": "2026-03-19T21:39:19.188199Z",
-     "iopub.status.idle": "2026-03-19T21:39:19.219554Z",
-     "shell.execute_reply": "2026-03-19T21:39:19.219284Z"
+     "iopub.execute_input": "2026-03-20T00:22:03.346971Z",
+     "iopub.status.busy": "2026-03-20T00:22:03.346852Z",
+     "iopub.status.idle": "2026-03-20T00:22:03.380005Z",
+     "shell.execute_reply": "2026-03-20T00:22:03.379523Z"
     }
    },
    "outputs": [],
diff --git a/nbs/hello_world_tutorial.ipynb b/nbs/hello_world_tutorial.ipynb
index 0002a5f..fde1441 100644
--- a/nbs/hello_world_tutorial.ipynb
+++ b/nbs/hello_world_tutorial.ipynb
@@ -50,10 +50,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:14.985893Z",
-     "iopub.status.busy": "2026-03-19T21:27:14.985409Z",
-     "iopub.status.idle": "2026-03-19T21:27:16.055040Z",
-     "shell.execute_reply": "2026-03-19T21:27:16.054736Z"
+     "iopub.execute_input": "2026-03-20T00:22:05.573352Z",
+     "iopub.status.busy": "2026-03-20T00:22:05.573181Z",
+     "iopub.status.idle": "2026-03-20T00:22:06.726737Z",
+     "shell.execute_reply": "2026-03-20T00:22:06.726344Z"
     }
    },
    "outputs": [],
@@ -78,10 +78,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:16.056764Z",
-     "iopub.status.busy": "2026-03-19T21:27:16.056632Z",
-     "iopub.status.idle": "2026-03-19T21:27:16.622700Z",
-     "shell.execute_reply": "2026-03-19T21:27:16.621455Z"
+     "iopub.execute_input": "2026-03-20T00:22:06.728687Z",
+     "iopub.status.busy": "2026-03-20T00:22:06.728514Z",
+     "iopub.status.idle": "2026-03-20T00:22:07.179179Z",
+     "shell.execute_reply": "2026-03-20T00:22:07.178834Z"
     }
    },
    "outputs": [],
@@ -134,10 +134,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:16.660364Z",
-     "iopub.status.busy": "2026-03-19T21:27:16.660174Z",
-     "iopub.status.idle": "2026-03-19T21:27:16.669562Z",
-     "shell.execute_reply": "2026-03-19T21:27:16.669240Z"
+     "iopub.execute_input": "2026-03-20T00:22:07.204932Z",
+     "iopub.status.busy": "2026-03-20T00:22:07.204681Z",
+     "iopub.status.idle": "2026-03-20T00:22:07.214478Z",
+     "shell.execute_reply": "2026-03-20T00:22:07.214140Z"
     }
    },
    "outputs": [],
@@ -152,10 +152,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:16.671301Z",
-     "iopub.status.busy": "2026-03-19T21:27:16.671121Z",
-     "iopub.status.idle": "2026-03-19T21:27:16.705482Z",
-     "shell.execute_reply": "2026-03-19T21:27:16.705208Z"
+     "iopub.execute_input": "2026-03-20T00:22:07.216177Z",
+     "iopub.status.busy": "2026-03-20T00:22:07.216038Z",
+     "iopub.status.idle": "2026-03-20T00:22:07.248817Z",
+     "shell.execute_reply": "2026-03-20T00:22:07.248514Z"
     }
    },
    "outputs": [],
@@ -180,10 +180,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:16.706950Z",
-     "iopub.status.busy": "2026-03-19T21:27:16.706840Z",
-     "iopub.status.idle": "2026-03-19T21:27:16.730461Z",
-     "shell.execute_reply": "2026-03-19T21:27:16.730172Z"
+     "iopub.execute_input": "2026-03-20T00:22:07.250439Z",
+     "iopub.status.busy": "2026-03-20T00:22:07.250305Z",
+     "iopub.status.idle": "2026-03-20T00:22:07.273848Z",
+     "shell.execute_reply": "2026-03-20T00:22:07.273454Z"
     }
    },
    "outputs": [],
@@ -210,10 +210,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:16.731864Z",
-     "iopub.status.busy": "2026-03-19T21:27:16.731743Z",
-     "iopub.status.idle": "2026-03-19T21:27:16.733939Z",
-     "shell.execute_reply": "2026-03-19T21:27:16.733692Z"
+     "iopub.execute_input": "2026-03-20T00:22:07.275365Z",
+     "iopub.status.busy": "2026-03-20T00:22:07.275250Z",
+     "iopub.status.idle": "2026-03-20T00:22:07.277471Z",
+     "shell.execute_reply": "2026-03-20T00:22:07.277224Z"
     }
    },
    "outputs": [],
@@ -235,10 +235,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:16.735312Z",
-     "iopub.status.busy": "2026-03-19T21:27:16.735207Z",
-     "iopub.status.idle": "2026-03-19T21:27:16.737191Z",
-     "shell.execute_reply": "2026-03-19T21:27:16.736837Z"
+     "iopub.execute_input": "2026-03-20T00:22:07.278821Z",
+     "iopub.status.busy": "2026-03-20T00:22:07.278715Z",
+     "iopub.status.idle": "2026-03-20T00:22:07.280717Z",
+     "shell.execute_reply": "2026-03-20T00:22:07.280418Z"
     }
    },
    "outputs": [],
@@ -266,10 +266,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:16.738832Z",
-     "iopub.status.busy": "2026-03-19T21:27:16.738674Z",
-     "iopub.status.idle": "2026-03-19T21:27:16.740692Z",
-     "shell.execute_reply": "2026-03-19T21:27:16.740447Z"
+     "iopub.execute_input": "2026-03-20T00:22:07.282150Z",
+     "iopub.status.busy": "2026-03-20T00:22:07.282040Z",
+     "iopub.status.idle": "2026-03-20T00:22:07.284074Z",
+     "shell.execute_reply": "2026-03-20T00:22:07.283783Z"
     }
    },
    "outputs": [],
@@ -291,10 +291,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:16.741995Z",
-     "iopub.status.busy": "2026-03-19T21:27:16.741889Z",
-     "iopub.status.idle": "2026-03-19T21:27:16.786433Z",
-     "shell.execute_reply": "2026-03-19T21:27:16.785961Z"
+     "iopub.execute_input": "2026-03-20T00:22:07.285705Z",
+     "iopub.status.busy": "2026-03-20T00:22:07.285496Z",
+     "iopub.status.idle": "2026-03-20T00:22:07.332915Z",
+     "shell.execute_reply": "2026-03-20T00:22:07.332524Z"
     }
    },
    "outputs": [],
diff --git a/nbs/homography.ipynb b/nbs/homography.ipynb
index f74d50d..6b0c419 100644
--- a/nbs/homography.ipynb
+++ b/nbs/homography.ipynb
@@ -41,10 +41,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:21.394708Z",
-     "iopub.status.busy": "2026-03-19T21:39:21.394292Z",
-     "iopub.status.idle": "2026-03-19T21:39:24.367952Z",
-     "shell.execute_reply": "2026-03-19T21:39:24.367103Z"
+     "iopub.execute_input": "2026-03-20T00:22:09.490680Z",
+     "iopub.status.busy": "2026-03-20T00:22:09.490504Z",
+     "iopub.status.idle": "2026-03-20T00:22:11.487009Z",
+     "shell.execute_reply": "2026-03-20T00:22:11.486544Z"
     }
    },
    "outputs": [],
@@ -82,10 +82,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:24.397138Z",
-     "iopub.status.busy": "2026-03-19T21:39:24.396931Z",
-     "iopub.status.idle": "2026-03-19T21:39:25.436815Z",
-     "shell.execute_reply": "2026-03-19T21:39:25.436495Z"
+     "iopub.execute_input": "2026-03-20T00:22:11.515787Z",
+     "iopub.status.busy": "2026-03-20T00:22:11.515605Z",
+     "iopub.status.idle": "2026-03-20T00:22:12.566491Z",
+     "shell.execute_reply": "2026-03-20T00:22:12.566032Z"
     }
    },
    "outputs": [],
@@ -119,10 +119,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:25.438627Z",
-     "iopub.status.busy": "2026-03-19T21:39:25.438452Z",
-     "iopub.status.idle": "2026-03-19T21:39:25.449736Z",
-     "shell.execute_reply": "2026-03-19T21:39:25.449453Z"
+     "iopub.execute_input": "2026-03-20T00:22:12.568510Z",
+     "iopub.status.busy": "2026-03-20T00:22:12.568328Z",
+     "iopub.status.idle": "2026-03-20T00:22:12.582461Z",
+     "shell.execute_reply": "2026-03-20T00:22:12.582188Z"
     }
    },
    "outputs": [],
@@ -133,7 +133,7 @@
     "error_tol: float = 1e-8  # the optimisation error tolerance\n",
     "\n",
     "log_interval: int = 100  # print log every N iterations\n",
-    "device = K.utils.get_cuda_or_mps_device_if_available()\n",
+    "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"mps\" if torch.backends.mps.is_available() else \"cpu\")\n",
     "print(\"Using \", device)"
    ]
   },
@@ -151,10 +151,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:25.451178Z",
-     "iopub.status.busy": "2026-03-19T21:39:25.451022Z",
-     "iopub.status.idle": "2026-03-19T21:39:25.453399Z",
-     "shell.execute_reply": "2026-03-19T21:39:25.453175Z"
+     "iopub.execute_input": "2026-03-20T00:22:12.583925Z",
+     "iopub.status.busy": "2026-03-20T00:22:12.583803Z",
+     "iopub.status.idle": "2026-03-20T00:22:12.586160Z",
+     "shell.execute_reply": "2026-03-20T00:22:12.585896Z"
     }
    },
    "outputs": [],
@@ -184,10 +184,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:25.454712Z",
-     "iopub.status.busy": "2026-03-19T21:39:25.454616Z",
-     "iopub.status.idle": "2026-03-19T21:39:25.727865Z",
-     "shell.execute_reply": "2026-03-19T21:39:25.727512Z"
+     "iopub.execute_input": "2026-03-20T00:22:12.587484Z",
+     "iopub.status.busy": "2026-03-20T00:22:12.587383Z",
+     "iopub.status.idle": "2026-03-20T00:22:12.833463Z",
+     "shell.execute_reply": "2026-03-20T00:22:12.833144Z"
     }
    },
    "outputs": [],
@@ -232,10 +232,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:25.729552Z",
-     "iopub.status.busy": "2026-03-19T21:39:25.729431Z",
-     "iopub.status.idle": "2026-03-19T21:39:25.903977Z",
-     "shell.execute_reply": "2026-03-19T21:39:25.903678Z"
+     "iopub.execute_input": "2026-03-20T00:22:12.835080Z",
+     "iopub.status.busy": "2026-03-20T00:22:12.834984Z",
+     "iopub.status.idle": "2026-03-20T00:22:13.008033Z",
+     "shell.execute_reply": "2026-03-20T00:22:13.007742Z"
     }
    },
    "outputs": [],
@@ -267,10 +267,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:25.907779Z",
-     "iopub.status.busy": "2026-03-19T21:39:25.907656Z",
-     "iopub.status.idle": "2026-03-19T21:39:26.361227Z",
-     "shell.execute_reply": "2026-03-19T21:39:26.360924Z"
+     "iopub.execute_input": "2026-03-20T00:22:13.011743Z",
+     "iopub.status.busy": "2026-03-20T00:22:13.011623Z",
+     "iopub.status.idle": "2026-03-20T00:22:13.467665Z",
+     "shell.execute_reply": "2026-03-20T00:22:13.467314Z"
     }
    },
    "outputs": [],
@@ -297,10 +297,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:26.363026Z",
-     "iopub.status.busy": "2026-03-19T21:39:26.362871Z",
-     "iopub.status.idle": "2026-03-19T21:39:26.461332Z",
-     "shell.execute_reply": "2026-03-19T21:39:26.461041Z"
+     "iopub.execute_input": "2026-03-20T00:22:13.469347Z",
+     "iopub.status.busy": "2026-03-20T00:22:13.469198Z",
+     "iopub.status.idle": "2026-03-20T00:22:13.508553Z",
+     "shell.execute_reply": "2026-03-20T00:22:13.508265Z"
     }
    },
    "outputs": [],
@@ -340,10 +340,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:26.462984Z",
-     "iopub.status.busy": "2026-03-19T21:39:26.462894Z",
-     "iopub.status.idle": "2026-03-19T21:39:26.466637Z",
-     "shell.execute_reply": "2026-03-19T21:39:26.466334Z"
+     "iopub.execute_input": "2026-03-20T00:22:13.510171Z",
+     "iopub.status.busy": "2026-03-20T00:22:13.510050Z",
+     "iopub.status.idle": "2026-03-20T00:22:13.513437Z",
+     "shell.execute_reply": "2026-03-20T00:22:13.513188Z"
     }
    },
    "outputs": [],
@@ -394,10 +394,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:39:26.468114Z",
-     "iopub.status.busy": "2026-03-19T21:39:26.468027Z",
-     "iopub.status.idle": "2026-03-19T21:40:10.148690Z",
-     "shell.execute_reply": "2026-03-19T21:40:10.148388Z"
+     "iopub.execute_input": "2026-03-20T00:22:13.514703Z",
+     "iopub.status.busy": "2026-03-20T00:22:13.514623Z",
+     "iopub.status.idle": "2026-03-20T00:22:26.080871Z",
+     "shell.execute_reply": "2026-03-20T00:22:26.080401Z"
     }
    },
    "outputs": [],
diff --git a/nbs/image_enhancement.ipynb b/nbs/image_enhancement.ipynb
index f72886e..375195c 100644
--- a/nbs/image_enhancement.ipynb
+++ b/nbs/image_enhancement.ipynb
@@ -40,10 +40,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:27.197464Z",
-     "iopub.status.busy": "2026-03-19T21:27:27.196822Z",
-     "iopub.status.idle": "2026-03-19T21:27:28.329935Z",
-     "shell.execute_reply": "2026-03-19T21:27:28.328170Z"
+     "iopub.execute_input": "2026-03-20T00:22:28.478283Z",
+     "iopub.status.busy": "2026-03-20T00:22:28.478140Z",
+     "iopub.status.idle": "2026-03-20T00:22:29.107669Z",
+     "shell.execute_reply": "2026-03-20T00:22:29.107247Z"
     }
    },
    "outputs": [],
@@ -72,10 +72,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:28.362001Z",
-     "iopub.status.busy": "2026-03-19T21:27:28.361771Z",
-     "iopub.status.idle": "2026-03-19T21:27:29.854481Z",
-     "shell.execute_reply": "2026-03-19T21:27:29.854184Z"
+     "iopub.execute_input": "2026-03-20T00:22:29.131094Z",
+     "iopub.status.busy": "2026-03-20T00:22:29.130909Z",
+     "iopub.status.idle": "2026-03-20T00:22:30.676310Z",
+     "shell.execute_reply": "2026-03-20T00:22:30.675957Z"
     }
    },
    "outputs": [],
@@ -94,10 +94,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:29.856085Z",
-     "iopub.status.busy": "2026-03-19T21:27:29.855949Z",
-     "iopub.status.idle": "2026-03-19T21:27:29.858111Z",
-     "shell.execute_reply": "2026-03-19T21:27:29.857856Z"
+     "iopub.execute_input": "2026-03-20T00:22:30.677979Z",
+     "iopub.status.busy": "2026-03-20T00:22:30.677834Z",
+     "iopub.status.idle": "2026-03-20T00:22:30.680165Z",
+     "shell.execute_reply": "2026-03-20T00:22:30.679894Z"
     }
    },
    "outputs": [],
@@ -123,15 +123,15 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:29.859415Z",
-     "iopub.status.busy": "2026-03-19T21:27:29.859326Z",
-     "iopub.status.idle": "2026-03-19T21:27:29.869023Z",
-     "shell.execute_reply": "2026-03-19T21:27:29.868766Z"
+     "iopub.execute_input": "2026-03-20T00:22:30.681653Z",
+     "iopub.status.busy": "2026-03-20T00:22:30.681548Z",
+     "iopub.status.idle": "2026-03-20T00:22:30.694106Z",
+     "shell.execute_reply": "2026-03-20T00:22:30.693811Z"
     }
    },
    "outputs": [],
    "source": [
-    "x_rgb = K.image_to_tensor(np.array(Image.open(\"ninja_turtles.jpg\").convert(\"RGB\")))[None, ...]"
+    "x_rgb = K.image_to_tensor(np.array(Image.open(\"ninja_turtles.jpg\").convert(\"RGB\"))).float()[None, ...] / 255.0"
    ]
   },
   {
@@ -147,10 +147,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:29.870399Z",
-     "iopub.status.busy": "2026-03-19T21:27:29.870311Z",
-     "iopub.status.idle": "2026-03-19T21:27:29.872123Z",
-     "shell.execute_reply": "2026-03-19T21:27:29.871894Z"
+     "iopub.execute_input": "2026-03-20T00:22:30.695816Z",
+     "iopub.status.busy": "2026-03-20T00:22:30.695712Z",
+     "iopub.status.idle": "2026-03-20T00:22:30.697642Z",
+     "shell.execute_reply": "2026-03-20T00:22:30.697321Z"
     }
    },
    "outputs": [],
@@ -163,10 +163,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:29.873303Z",
-     "iopub.status.busy": "2026-03-19T21:27:29.873221Z",
-     "iopub.status.idle": "2026-03-19T21:27:29.955342Z",
-     "shell.execute_reply": "2026-03-19T21:27:29.955030Z"
+     "iopub.execute_input": "2026-03-20T00:22:30.698964Z",
+     "iopub.status.busy": "2026-03-20T00:22:30.698871Z",
+     "iopub.status.idle": "2026-03-20T00:22:30.790529Z",
+     "shell.execute_reply": "2026-03-20T00:22:30.790195Z"
     }
    },
    "outputs": [],
@@ -186,10 +186,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:29.957385Z",
-     "iopub.status.busy": "2026-03-19T21:27:29.957293Z",
-     "iopub.status.idle": "2026-03-19T21:27:30.045742Z",
-     "shell.execute_reply": "2026-03-19T21:27:30.045424Z"
+     "iopub.execute_input": "2026-03-20T00:22:30.792678Z",
+     "iopub.status.busy": "2026-03-20T00:22:30.792583Z",
+     "iopub.status.idle": "2026-03-20T00:22:30.872901Z",
+     "shell.execute_reply": "2026-03-20T00:22:30.872611Z"
     }
    },
    "outputs": [],
@@ -210,10 +210,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:30.047192Z",
-     "iopub.status.busy": "2026-03-19T21:27:30.047080Z",
-     "iopub.status.idle": "2026-03-19T21:27:30.120546Z",
-     "shell.execute_reply": "2026-03-19T21:27:30.120287Z"
+     "iopub.execute_input": "2026-03-20T00:22:30.874846Z",
+     "iopub.status.busy": "2026-03-20T00:22:30.874729Z",
+     "iopub.status.idle": "2026-03-20T00:22:30.958292Z",
+     "shell.execute_reply": "2026-03-20T00:22:30.958006Z"
     }
    },
    "outputs": [],
@@ -234,10 +234,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:30.121943Z",
-     "iopub.status.busy": "2026-03-19T21:27:30.121835Z",
-     "iopub.status.idle": "2026-03-19T21:27:30.272282Z",
-     "shell.execute_reply": "2026-03-19T21:27:30.271982Z"
+     "iopub.execute_input": "2026-03-20T00:22:30.960231Z",
+     "iopub.status.busy": "2026-03-20T00:22:30.960118Z",
+     "iopub.status.idle": "2026-03-20T00:22:31.117966Z",
+     "shell.execute_reply": "2026-03-20T00:22:31.117665Z"
     }
    },
    "outputs": [],
@@ -258,10 +258,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:30.273721Z",
-     "iopub.status.busy": "2026-03-19T21:27:30.273610Z",
-     "iopub.status.idle": "2026-03-19T21:27:30.346726Z",
-     "shell.execute_reply": "2026-03-19T21:27:30.346473Z"
+     "iopub.execute_input": "2026-03-20T00:22:31.119717Z",
+     "iopub.status.busy": "2026-03-20T00:22:31.119609Z",
+     "iopub.status.idle": "2026-03-20T00:22:31.212807Z",
+     "shell.execute_reply": "2026-03-20T00:22:31.212512Z"
     }
    },
    "outputs": [],
@@ -282,10 +282,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:27:30.348135Z",
-     "iopub.status.busy": "2026-03-19T21:27:30.348027Z",
-     "iopub.status.idle": "2026-03-19T21:27:30.477066Z",
-     "shell.execute_reply": "2026-03-19T21:27:30.476797Z"
+     "iopub.execute_input": "2026-03-20T00:22:31.214714Z",
+     "iopub.status.busy": "2026-03-20T00:22:31.214612Z",
+     "iopub.status.idle": "2026-03-20T00:22:31.338861Z",
+     "shell.execute_reply": "2026-03-20T00:22:31.338589Z"
     }
    },
    "outputs": [],
diff --git a/nbs/image_histogram.ipynb b/nbs/image_histogram.ipynb
index 0875314..d7c2763 100644
--- a/nbs/image_histogram.ipynb
+++ b/nbs/image_histogram.ipynb
@@ -56,10 +56,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:12.549886Z",
-     "iopub.status.busy": "2026-03-19T21:40:12.549320Z",
-     "iopub.status.idle": "2026-03-19T21:40:14.588576Z",
-     "shell.execute_reply": "2026-03-19T21:40:14.586731Z"
+     "iopub.execute_input": "2026-03-20T00:22:33.663955Z",
+     "iopub.status.busy": "2026-03-20T00:22:33.663635Z",
+     "iopub.status.idle": "2026-03-20T00:22:35.472246Z",
+     "shell.execute_reply": "2026-03-20T00:22:35.471863Z"
     }
    },
    "outputs": [],
@@ -90,10 +90,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:14.613934Z",
-     "iopub.status.busy": "2026-03-19T21:40:14.613746Z",
-     "iopub.status.idle": "2026-03-19T21:40:16.183986Z",
-     "shell.execute_reply": "2026-03-19T21:40:16.183649Z"
+     "iopub.execute_input": "2026-03-20T00:22:35.495198Z",
+     "iopub.status.busy": "2026-03-20T00:22:35.495017Z",
+     "iopub.status.idle": "2026-03-20T00:22:37.120858Z",
+     "shell.execute_reply": "2026-03-20T00:22:37.120515Z"
     }
    },
    "outputs": [],
@@ -122,10 +122,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:16.185863Z",
-     "iopub.status.busy": "2026-03-19T21:40:16.185696Z",
-     "iopub.status.idle": "2026-03-19T21:40:16.188053Z",
-     "shell.execute_reply": "2026-03-19T21:40:16.187795Z"
+     "iopub.execute_input": "2026-03-20T00:22:37.122931Z",
+     "iopub.status.busy": "2026-03-20T00:22:37.122726Z",
+     "iopub.status.idle": "2026-03-20T00:22:37.125249Z",
+     "shell.execute_reply": "2026-03-20T00:22:37.124957Z"
     }
    },
    "outputs": [],
@@ -150,10 +150,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:16.189508Z",
-     "iopub.status.busy": "2026-03-19T21:40:16.189389Z",
-     "iopub.status.idle": "2026-03-19T21:40:16.191743Z",
-     "shell.execute_reply": "2026-03-19T21:40:16.191480Z"
+     "iopub.execute_input": "2026-03-20T00:22:37.126660Z",
+     "iopub.status.busy": "2026-03-20T00:22:37.126555Z",
+     "iopub.status.idle": "2026-03-20T00:22:37.128647Z",
+     "shell.execute_reply": "2026-03-20T00:22:37.128387Z"
     }
    },
    "outputs": [],
@@ -178,10 +178,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:16.193058Z",
-     "iopub.status.busy": "2026-03-19T21:40:16.192950Z",
-     "iopub.status.idle": "2026-03-19T21:40:16.195801Z",
-     "shell.execute_reply": "2026-03-19T21:40:16.195552Z"
+     "iopub.execute_input": "2026-03-20T00:22:37.129998Z",
+     "iopub.status.busy": "2026-03-20T00:22:37.129900Z",
+     "iopub.status.idle": "2026-03-20T00:22:37.132838Z",
+     "shell.execute_reply": "2026-03-20T00:22:37.132559Z"
     }
    },
    "outputs": [],
@@ -220,10 +220,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:16.197126Z",
-     "iopub.status.busy": "2026-03-19T21:40:16.197021Z",
-     "iopub.status.idle": "2026-03-19T21:40:16.394855Z",
-     "shell.execute_reply": "2026-03-19T21:40:16.394582Z"
+     "iopub.execute_input": "2026-03-20T00:22:37.134135Z",
+     "iopub.status.busy": "2026-03-20T00:22:37.134033Z",
+     "iopub.status.idle": "2026-03-20T00:22:37.335462Z",
+     "shell.execute_reply": "2026-03-20T00:22:37.335098Z"
     }
    },
    "outputs": [],
@@ -283,10 +283,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:16.397585Z",
-     "iopub.status.busy": "2026-03-19T21:40:16.397461Z",
-     "iopub.status.idle": "2026-03-19T21:40:16.399263Z",
-     "shell.execute_reply": "2026-03-19T21:40:16.399040Z"
+     "iopub.execute_input": "2026-03-20T00:22:37.338270Z",
+     "iopub.status.busy": "2026-03-20T00:22:37.338115Z",
+     "iopub.status.idle": "2026-03-20T00:22:37.340031Z",
+     "shell.execute_reply": "2026-03-20T00:22:37.339752Z"
     }
    },
    "outputs": [],
@@ -306,10 +306,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:16.400652Z",
-     "iopub.status.busy": "2026-03-19T21:40:16.400512Z",
-     "iopub.status.idle": "2026-03-19T21:40:16.534490Z",
-     "shell.execute_reply": "2026-03-19T21:40:16.534155Z"
+     "iopub.execute_input": "2026-03-20T00:22:37.341417Z",
+     "iopub.status.busy": "2026-03-20T00:22:37.341301Z",
+     "iopub.status.idle": "2026-03-20T00:22:37.480240Z",
+     "shell.execute_reply": "2026-03-20T00:22:37.479933Z"
     }
    },
    "outputs": [],
@@ -337,10 +337,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:16.536057Z",
-     "iopub.status.busy": "2026-03-19T21:40:16.535927Z",
-     "iopub.status.idle": "2026-03-19T21:40:16.654676Z",
-     "shell.execute_reply": "2026-03-19T21:40:16.654323Z"
+     "iopub.execute_input": "2026-03-20T00:22:37.481661Z",
+     "iopub.status.busy": "2026-03-20T00:22:37.481542Z",
+     "iopub.status.idle": "2026-03-20T00:22:37.598166Z",
+     "shell.execute_reply": "2026-03-20T00:22:37.597880Z"
     }
    },
    "outputs": [],
@@ -375,10 +375,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:16.656721Z",
-     "iopub.status.busy": "2026-03-19T21:40:16.656615Z",
-     "iopub.status.idle": "2026-03-19T21:40:16.738289Z",
-     "shell.execute_reply": "2026-03-19T21:40:16.738004Z"
+     "iopub.execute_input": "2026-03-20T00:22:37.600143Z",
+     "iopub.status.busy": "2026-03-20T00:22:37.600040Z",
+     "iopub.status.idle": "2026-03-20T00:22:37.681495Z",
+     "shell.execute_reply": "2026-03-20T00:22:37.681226Z"
     }
    },
    "outputs": [],
@@ -399,10 +399,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:16.739784Z",
-     "iopub.status.busy": "2026-03-19T21:40:16.739670Z",
-     "iopub.status.idle": "2026-03-19T21:40:16.909174Z",
-     "shell.execute_reply": "2026-03-19T21:40:16.908877Z"
+     "iopub.execute_input": "2026-03-20T00:22:37.682915Z",
+     "iopub.status.busy": "2026-03-20T00:22:37.682806Z",
+     "iopub.status.idle": "2026-03-20T00:22:37.851044Z",
+     "shell.execute_reply": "2026-03-20T00:22:37.850671Z"
     }
    },
    "outputs": [],
@@ -425,10 +425,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:16.911030Z",
-     "iopub.status.busy": "2026-03-19T21:40:16.910939Z",
-     "iopub.status.idle": "2026-03-19T21:40:17.040044Z",
-     "shell.execute_reply": "2026-03-19T21:40:17.039746Z"
+     "iopub.execute_input": "2026-03-20T00:22:37.853988Z",
+     "iopub.status.busy": "2026-03-20T00:22:37.853777Z",
+     "iopub.status.idle": "2026-03-20T00:22:37.980625Z",
+     "shell.execute_reply": "2026-03-20T00:22:37.980347Z"
     }
    },
    "outputs": [],
@@ -460,10 +460,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:17.042472Z",
-     "iopub.status.busy": "2026-03-19T21:40:17.042362Z",
-     "iopub.status.idle": "2026-03-19T21:40:17.132666Z",
-     "shell.execute_reply": "2026-03-19T21:40:17.132361Z"
+     "iopub.execute_input": "2026-03-20T00:22:37.982799Z",
+     "iopub.status.busy": "2026-03-20T00:22:37.982686Z",
+     "iopub.status.idle": "2026-03-20T00:22:38.073810Z",
+     "shell.execute_reply": "2026-03-20T00:22:38.073535Z"
     }
    },
    "outputs": [],
@@ -484,10 +484,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:17.134331Z",
-     "iopub.status.busy": "2026-03-19T21:40:17.134187Z",
-     "iopub.status.idle": "2026-03-19T21:40:17.291637Z",
-     "shell.execute_reply": "2026-03-19T21:40:17.291360Z"
+     "iopub.execute_input": "2026-03-20T00:22:38.075302Z",
+     "iopub.status.busy": "2026-03-20T00:22:38.075196Z",
+     "iopub.status.idle": "2026-03-20T00:22:38.234525Z",
+     "shell.execute_reply": "2026-03-20T00:22:38.234234Z"
     }
    },
    "outputs": [],
@@ -508,10 +508,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:17.293576Z",
-     "iopub.status.busy": "2026-03-19T21:40:17.293489Z",
-     "iopub.status.idle": "2026-03-19T21:40:17.441775Z",
-     "shell.execute_reply": "2026-03-19T21:40:17.441482Z"
+     "iopub.execute_input": "2026-03-20T00:22:38.237135Z",
+     "iopub.status.busy": "2026-03-20T00:22:38.236998Z",
+     "iopub.status.idle": "2026-03-20T00:22:38.381017Z",
+     "shell.execute_reply": "2026-03-20T00:22:38.380677Z"
     }
    },
    "outputs": [],
@@ -534,10 +534,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:17.443980Z",
-     "iopub.status.busy": "2026-03-19T21:40:17.443888Z",
-     "iopub.status.idle": "2026-03-19T21:40:17.585115Z",
-     "shell.execute_reply": "2026-03-19T21:40:17.584690Z"
+     "iopub.execute_input": "2026-03-20T00:22:38.383086Z",
+     "iopub.status.busy": "2026-03-20T00:22:38.382992Z",
+     "iopub.status.idle": "2026-03-20T00:22:38.524970Z",
+     "shell.execute_reply": "2026-03-20T00:22:38.524666Z"
     }
    },
    "outputs": [],
@@ -577,10 +577,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:17.587170Z",
-     "iopub.status.busy": "2026-03-19T21:40:17.587060Z",
-     "iopub.status.idle": "2026-03-19T21:40:17.776608Z",
-     "shell.execute_reply": "2026-03-19T21:40:17.776305Z"
+     "iopub.execute_input": "2026-03-20T00:22:38.527038Z",
+     "iopub.status.busy": "2026-03-20T00:22:38.526944Z",
+     "iopub.status.idle": "2026-03-20T00:22:38.710156Z",
+     "shell.execute_reply": "2026-03-20T00:22:38.709737Z"
     }
    },
    "outputs": [],
@@ -601,10 +601,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:17.778332Z",
-     "iopub.status.busy": "2026-03-19T21:40:17.778202Z",
-     "iopub.status.idle": "2026-03-19T21:40:17.948292Z",
-     "shell.execute_reply": "2026-03-19T21:40:17.947982Z"
+     "iopub.execute_input": "2026-03-20T00:22:38.711795Z",
+     "iopub.status.busy": "2026-03-20T00:22:38.711707Z",
+     "iopub.status.idle": "2026-03-20T00:22:38.874282Z",
+     "shell.execute_reply": "2026-03-20T00:22:38.873894Z"
     }
    },
    "outputs": [],
@@ -625,10 +625,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:17.950728Z",
-     "iopub.status.busy": "2026-03-19T21:40:17.950618Z",
-     "iopub.status.idle": "2026-03-19T21:40:18.158283Z",
-     "shell.execute_reply": "2026-03-19T21:40:18.157953Z"
+     "iopub.execute_input": "2026-03-20T00:22:38.876932Z",
+     "iopub.status.busy": "2026-03-20T00:22:38.876770Z",
+     "iopub.status.idle": "2026-03-20T00:22:39.091113Z",
+     "shell.execute_reply": "2026-03-20T00:22:39.090804Z"
     }
    },
    "outputs": [],
@@ -658,10 +658,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:18.160601Z",
-     "iopub.status.busy": "2026-03-19T21:40:18.160508Z",
-     "iopub.status.idle": "2026-03-19T21:40:18.274561Z",
-     "shell.execute_reply": "2026-03-19T21:40:18.274287Z"
+     "iopub.execute_input": "2026-03-20T00:22:39.094234Z",
+     "iopub.status.busy": "2026-03-20T00:22:39.094099Z",
+     "iopub.status.idle": "2026-03-20T00:22:39.206999Z",
+     "shell.execute_reply": "2026-03-20T00:22:39.206679Z"
     }
    },
    "outputs": [],
@@ -682,10 +682,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:18.276101Z",
-     "iopub.status.busy": "2026-03-19T21:40:18.275996Z",
-     "iopub.status.idle": "2026-03-19T21:40:18.438506Z",
-     "shell.execute_reply": "2026-03-19T21:40:18.438213Z"
+     "iopub.execute_input": "2026-03-20T00:22:39.208633Z",
+     "iopub.status.busy": "2026-03-20T00:22:39.208521Z",
+     "iopub.status.idle": "2026-03-20T00:22:39.361169Z",
+     "shell.execute_reply": "2026-03-20T00:22:39.360812Z"
     }
    },
    "outputs": [],
@@ -713,10 +713,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:18.440566Z",
-     "iopub.status.busy": "2026-03-19T21:40:18.440472Z",
-     "iopub.status.idle": "2026-03-19T21:40:18.630831Z",
-     "shell.execute_reply": "2026-03-19T21:40:18.630527Z"
+     "iopub.execute_input": "2026-03-20T00:22:39.363744Z",
+     "iopub.status.busy": "2026-03-20T00:22:39.363644Z",
+     "iopub.status.idle": "2026-03-20T00:22:39.550140Z",
+     "shell.execute_reply": "2026-03-20T00:22:39.549852Z"
     }
    },
    "outputs": [],
@@ -737,10 +737,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:18.633720Z",
-     "iopub.status.busy": "2026-03-19T21:40:18.633610Z",
-     "iopub.status.idle": "2026-03-19T21:40:18.765582Z",
-     "shell.execute_reply": "2026-03-19T21:40:18.765254Z"
+     "iopub.execute_input": "2026-03-20T00:22:39.552715Z",
+     "iopub.status.busy": "2026-03-20T00:22:39.552606Z",
+     "iopub.status.idle": "2026-03-20T00:22:39.686268Z",
+     "shell.execute_reply": "2026-03-20T00:22:39.685899Z"
     }
    },
    "outputs": [],
diff --git a/nbs/image_matching.ipynb b/nbs/image_matching.ipynb
index a9b44b5..33a0393 100644
--- a/nbs/image_matching.ipynb
+++ b/nbs/image_matching.ipynb
@@ -64,7 +64,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:22:42.051153Z",
+     "iopub.status.busy": "2026-03-20T00:22:42.051045Z",
+     "iopub.status.idle": "2026-03-20T00:22:42.091614Z",
+     "shell.execute_reply": "2026-03-20T00:22:42.091245Z"
+    }
+   },
    "outputs": [],
    "source": [
     "import io\n",
@@ -86,7 +93,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:22:42.093505Z",
+     "iopub.status.busy": "2026-03-20T00:22:42.093368Z",
+     "iopub.status.idle": "2026-03-20T00:22:42.609104Z",
+     "shell.execute_reply": "2026-03-20T00:22:42.608802Z"
+    }
+   },
    "outputs": [],
    "source": [
     "url_a = \"https://github.com/kornia/data/raw/main/matching/kn_church-2.jpg\"\n",
@@ -106,7 +120,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:22:42.632797Z",
+     "iopub.status.busy": "2026-03-20T00:22:42.632625Z",
+     "iopub.status.idle": "2026-03-20T00:22:43.698353Z",
+     "shell.execute_reply": "2026-03-20T00:22:43.698050Z"
+    }
+   },
    "outputs": [],
    "source": [
     "import cv2\n",
@@ -122,15 +143,22 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:22:43.699966Z",
+     "iopub.status.busy": "2026-03-20T00:22:43.699835Z",
+     "iopub.status.idle": "2026-03-20T00:22:45.221755Z",
+     "shell.execute_reply": "2026-03-20T00:22:45.221346Z"
+    }
+   },
    "outputs": [],
    "source": [
     "%%capture\n",
     "fname1 = \"kn_church-2.jpg\"\n",
     "fname2 = \"kn_church-8.jpg\"\n",
     "\n",
-    "img1 = K.image_to_tensor(np.array(Image.open(fname1).convert(\"RGB\")))[None, ...]\n",
-    "img2 = K.image_to_tensor(np.array(Image.open(fname2).convert(\"RGB\")))[None, ...]\n",
+    "img1 = K.image_to_tensor(np.array(Image.open(fname1).convert(\"RGB\"))).float()[None, ...] / 255.0\n",
+    "img2 = K.image_to_tensor(np.array(Image.open(fname2).convert(\"RGB\"))).float()[None, ...] / 255.0\n",
     "\n",
     "img1 = K.geometry.resize(img1, (600, 375), antialias=True)\n",
     "img2 = K.geometry.resize(img2, (600, 375), antialias=True)\n",
@@ -150,7 +178,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:22:45.223660Z",
+     "iopub.status.busy": "2026-03-20T00:22:45.223524Z",
+     "iopub.status.idle": "2026-03-20T00:22:45.225556Z",
+     "shell.execute_reply": "2026-03-20T00:22:45.225315Z"
+    }
+   },
    "outputs": [],
    "source": [
     "for k, v in correspondences.items():\n",
@@ -167,7 +202,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:22:45.226885Z",
+     "iopub.status.busy": "2026-03-20T00:22:45.226783Z",
+     "iopub.status.idle": "2026-03-20T00:22:45.233873Z",
+     "shell.execute_reply": "2026-03-20T00:22:45.233635Z"
+    }
+   },
    "outputs": [],
    "source": [
     "mkpts0 = correspondences[\"keypoints0\"].cpu().numpy()\n",
@@ -186,7 +228,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:22:45.235279Z",
+     "iopub.status.busy": "2026-03-20T00:22:45.235183Z",
+     "iopub.status.idle": "2026-03-20T00:22:45.849282Z",
+     "shell.execute_reply": "2026-03-20T00:22:45.848858Z"
+    }
+   },
    "outputs": [],
    "source": [
     "draw_LAF_matches(\n",
@@ -225,7 +274,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:22:45.856982Z",
+     "iopub.status.busy": "2026-03-20T00:22:45.856818Z",
+     "iopub.status.idle": "2026-03-20T00:22:47.127369Z",
+     "shell.execute_reply": "2026-03-20T00:22:47.127067Z"
+    }
+   },
    "outputs": [],
    "source": [
     "%%capture\n",
@@ -239,14 +295,21 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:22:47.128910Z",
+     "iopub.status.busy": "2026-03-20T00:22:47.128818Z",
+     "iopub.status.idle": "2026-03-20T00:22:49.675145Z",
+     "shell.execute_reply": "2026-03-20T00:22:49.674831Z"
+    }
+   },
    "outputs": [],
    "source": [
     "fname1 = \"scene0711_00_frame-001680.jpg\"\n",
     "fname2 = \"scene0711_00_frame-001995.jpg\"\n",
     "\n",
-    "img1 = K.image_to_tensor(np.array(Image.open(fname1).convert(\"RGB\")))[None, ...]\n",
-    "img2 = K.image_to_tensor(np.array(Image.open(fname2).convert(\"RGB\")))[None, ...]\n",
+    "img1 = K.image_to_tensor(np.array(Image.open(fname1).convert(\"RGB\"))).float()[None, ...] / 255.0\n",
+    "img2 = K.image_to_tensor(np.array(Image.open(fname2).convert(\"RGB\"))).float()[None, ...] / 255.0\n",
     "\n",
     "img1 = K.geometry.resize(img1, (480, 640), antialias=True)\n",
     "img2 = K.geometry.resize(img2, (480, 640), antialias=True)\n",
@@ -296,6 +359,18 @@
    "display_name": "python3",
    "language": "python",
    "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.15"
   }
  },
  "nbformat": 4,
diff --git a/nbs/image_matching_adalam.ipynb b/nbs/image_matching_adalam.ipynb
index b12454b..34849ad 100644
--- a/nbs/image_matching_adalam.ipynb
+++ b/nbs/image_matching_adalam.ipynb
@@ -68,10 +68,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:21.033342Z",
-     "iopub.status.busy": "2026-03-19T21:40:21.031712Z",
-     "iopub.status.idle": "2026-03-19T21:40:22.003739Z",
-     "shell.execute_reply": "2026-03-19T21:40:22.002764Z"
+     "iopub.execute_input": "2026-03-20T00:22:51.922743Z",
+     "iopub.status.busy": "2026-03-20T00:22:51.922459Z",
+     "iopub.status.idle": "2026-03-20T00:22:52.431386Z",
+     "shell.execute_reply": "2026-03-20T00:22:52.431041Z"
     }
    },
    "outputs": [],
@@ -110,10 +110,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:22.030917Z",
-     "iopub.status.busy": "2026-03-19T21:40:22.030731Z",
-     "iopub.status.idle": "2026-03-19T21:40:23.132519Z",
-     "shell.execute_reply": "2026-03-19T21:40:23.132220Z"
+     "iopub.execute_input": "2026-03-20T00:22:52.455563Z",
+     "iopub.status.busy": "2026-03-20T00:22:52.455384Z",
+     "iopub.status.idle": "2026-03-20T00:22:53.534768Z",
+     "shell.execute_reply": "2026-03-20T00:22:53.534410Z"
     }
    },
    "outputs": [],
@@ -137,10 +137,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:23.134414Z",
-     "iopub.status.busy": "2026-03-19T21:40:23.134232Z",
-     "iopub.status.idle": "2026-03-19T21:40:40.962408Z",
-     "shell.execute_reply": "2026-03-19T21:40:40.962057Z"
+     "iopub.execute_input": "2026-03-20T00:22:53.536695Z",
+     "iopub.status.busy": "2026-03-20T00:22:53.536501Z",
+     "iopub.status.idle": "2026-03-20T00:23:10.730429Z",
+     "shell.execute_reply": "2026-03-20T00:23:10.730086Z"
     }
    },
    "outputs": [],
@@ -183,10 +183,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:40.964135Z",
-     "iopub.status.busy": "2026-03-19T21:40:40.964009Z",
-     "iopub.status.idle": "2026-03-19T21:40:40.966143Z",
-     "shell.execute_reply": "2026-03-19T21:40:40.965899Z"
+     "iopub.execute_input": "2026-03-20T00:23:10.732043Z",
+     "iopub.status.busy": "2026-03-20T00:23:10.731914Z",
+     "iopub.status.idle": "2026-03-20T00:23:10.734002Z",
+     "shell.execute_reply": "2026-03-20T00:23:10.733767Z"
     }
    },
    "outputs": [],
@@ -199,10 +199,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:40.967443Z",
-     "iopub.status.busy": "2026-03-19T21:40:40.967346Z",
-     "iopub.status.idle": "2026-03-19T21:40:40.977855Z",
-     "shell.execute_reply": "2026-03-19T21:40:40.977580Z"
+     "iopub.execute_input": "2026-03-20T00:23:10.735298Z",
+     "iopub.status.busy": "2026-03-20T00:23:10.735216Z",
+     "iopub.status.idle": "2026-03-20T00:23:10.745290Z",
+     "shell.execute_reply": "2026-03-20T00:23:10.745052Z"
     }
    },
    "outputs": [],
@@ -232,10 +232,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:40.979244Z",
-     "iopub.status.busy": "2026-03-19T21:40:40.979151Z",
-     "iopub.status.idle": "2026-03-19T21:40:41.614811Z",
-     "shell.execute_reply": "2026-03-19T21:40:41.614307Z"
+     "iopub.execute_input": "2026-03-20T00:23:10.746727Z",
+     "iopub.status.busy": "2026-03-20T00:23:10.746644Z",
+     "iopub.status.idle": "2026-03-20T00:23:11.379217Z",
+     "shell.execute_reply": "2026-03-20T00:23:11.378650Z"
     }
    },
    "outputs": [],
diff --git a/nbs/image_matching_disk.ipynb b/nbs/image_matching_disk.ipynb
index a247c57..b4f47e7 100644
--- a/nbs/image_matching_disk.ipynb
+++ b/nbs/image_matching_disk.ipynb
@@ -67,10 +67,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:44.049307Z",
-     "iopub.status.busy": "2026-03-19T21:40:44.048834Z",
-     "iopub.status.idle": "2026-03-19T21:40:44.583636Z",
-     "shell.execute_reply": "2026-03-19T21:40:44.582673Z"
+     "iopub.execute_input": "2026-03-20T00:23:13.860999Z",
+     "iopub.status.busy": "2026-03-20T00:23:13.860870Z",
+     "iopub.status.idle": "2026-03-20T00:23:14.405136Z",
+     "shell.execute_reply": "2026-03-20T00:23:14.404651Z"
     }
    },
    "outputs": [],
@@ -109,10 +109,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:44.617846Z",
-     "iopub.status.busy": "2026-03-19T21:40:44.617662Z",
-     "iopub.status.idle": "2026-03-19T21:40:45.799555Z",
-     "shell.execute_reply": "2026-03-19T21:40:45.799251Z"
+     "iopub.execute_input": "2026-03-20T00:23:14.429330Z",
+     "iopub.status.busy": "2026-03-20T00:23:14.429147Z",
+     "iopub.status.idle": "2026-03-20T00:23:15.640270Z",
+     "shell.execute_reply": "2026-03-20T00:23:15.639898Z"
     }
    },
    "outputs": [],
@@ -128,7 +128,7 @@
     "from kornia_moons.viz import *\n",
     "from PIL import Image\n",
     "\n",
-    "device = K.utils.get_cuda_or_mps_device_if_available()\n",
+    "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"mps\" if torch.backends.mps.is_available() else \"cpu\")\n",
     "print(device)"
    ]
   },
@@ -137,10 +137,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:45.801012Z",
-     "iopub.status.busy": "2026-03-19T21:40:45.800843Z",
-     "iopub.status.idle": "2026-03-19T21:40:50.728228Z",
-     "shell.execute_reply": "2026-03-19T21:40:50.727950Z"
+     "iopub.execute_input": "2026-03-20T00:23:15.642082Z",
+     "iopub.status.busy": "2026-03-20T00:23:15.641906Z",
+     "iopub.status.idle": "2026-03-20T00:23:16.509417Z",
+     "shell.execute_reply": "2026-03-20T00:23:16.509099Z"
     }
    },
    "outputs": [],
@@ -190,10 +190,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:50.729650Z",
-     "iopub.status.busy": "2026-03-19T21:40:50.729561Z",
-     "iopub.status.idle": "2026-03-19T21:40:50.740802Z",
-     "shell.execute_reply": "2026-03-19T21:40:50.740507Z"
+     "iopub.execute_input": "2026-03-20T00:23:16.510905Z",
+     "iopub.status.busy": "2026-03-20T00:23:16.510756Z",
+     "iopub.status.idle": "2026-03-20T00:23:16.521453Z",
+     "shell.execute_reply": "2026-03-20T00:23:16.521185Z"
     }
    },
    "outputs": [],
@@ -230,10 +230,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:50.742316Z",
-     "iopub.status.busy": "2026-03-19T21:40:50.742155Z",
-     "iopub.status.idle": "2026-03-19T21:40:51.213141Z",
-     "shell.execute_reply": "2026-03-19T21:40:51.212718Z"
+     "iopub.execute_input": "2026-03-20T00:23:16.523051Z",
+     "iopub.status.busy": "2026-03-20T00:23:16.522942Z",
+     "iopub.status.idle": "2026-03-20T00:23:16.984961Z",
+     "shell.execute_reply": "2026-03-20T00:23:16.984382Z"
     }
    },
    "outputs": [],
diff --git a/nbs/image_matching_lightglue.ipynb b/nbs/image_matching_lightglue.ipynb
index 87ede7f..b92cedc 100644
--- a/nbs/image_matching_lightglue.ipynb
+++ b/nbs/image_matching_lightglue.ipynb
@@ -66,10 +66,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:53.641598Z",
-     "iopub.status.busy": "2026-03-19T21:40:53.641175Z",
-     "iopub.status.idle": "2026-03-19T21:40:54.367205Z",
-     "shell.execute_reply": "2026-03-19T21:40:54.365940Z"
+     "iopub.execute_input": "2026-03-20T00:23:19.388491Z",
+     "iopub.status.busy": "2026-03-20T00:23:19.388217Z",
+     "iopub.status.idle": "2026-03-20T00:23:19.950738Z",
+     "shell.execute_reply": "2026-03-20T00:23:19.950404Z"
     }
    },
    "outputs": [],
@@ -108,10 +108,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:54.395235Z",
-     "iopub.status.busy": "2026-03-19T21:40:54.395015Z",
-     "iopub.status.idle": "2026-03-19T21:40:55.642136Z",
-     "shell.execute_reply": "2026-03-19T21:40:55.641867Z"
+     "iopub.execute_input": "2026-03-20T00:23:19.977628Z",
+     "iopub.status.busy": "2026-03-20T00:23:19.977419Z",
+     "iopub.status.idle": "2026-03-20T00:23:21.057105Z",
+     "shell.execute_reply": "2026-03-20T00:23:21.056776Z"
     }
    },
    "outputs": [],
@@ -127,7 +127,7 @@
     "from kornia_moons.viz import *\n",
     "from PIL import Image\n",
     "\n",
-    "device = K.utils.get_cuda_or_mps_device_if_available()\n",
+    "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"mps\" if torch.backends.mps.is_available() else \"cpu\")\n",
     "print(device)"
    ]
   },
@@ -143,10 +143,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:40:55.643562Z",
-     "iopub.status.busy": "2026-03-19T21:40:55.643388Z",
-     "iopub.status.idle": "2026-03-19T21:41:02.464152Z",
-     "shell.execute_reply": "2026-03-19T21:41:02.463847Z"
+     "iopub.execute_input": "2026-03-20T00:23:21.058630Z",
+     "iopub.status.busy": "2026-03-20T00:23:21.058473Z",
+     "iopub.status.idle": "2026-03-20T00:23:22.984555Z",
+     "shell.execute_reply": "2026-03-20T00:23:22.984261Z"
     }
    },
    "outputs": [],
@@ -193,10 +193,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:41:02.465572Z",
-     "iopub.status.busy": "2026-03-19T21:41:02.465461Z",
-     "iopub.status.idle": "2026-03-19T21:41:02.848996Z",
-     "shell.execute_reply": "2026-03-19T21:41:02.848664Z"
+     "iopub.execute_input": "2026-03-20T00:23:22.985947Z",
+     "iopub.status.busy": "2026-03-20T00:23:22.985844Z",
+     "iopub.status.idle": "2026-03-20T00:23:23.318057Z",
+     "shell.execute_reply": "2026-03-20T00:23:23.317743Z"
     }
    },
    "outputs": [],
@@ -232,10 +232,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:41:02.850526Z",
-     "iopub.status.busy": "2026-03-19T21:41:02.850417Z",
-     "iopub.status.idle": "2026-03-19T21:41:02.855807Z",
-     "shell.execute_reply": "2026-03-19T21:41:02.855530Z"
+     "iopub.execute_input": "2026-03-20T00:23:23.319573Z",
+     "iopub.status.busy": "2026-03-20T00:23:23.319454Z",
+     "iopub.status.idle": "2026-03-20T00:23:23.324841Z",
+     "shell.execute_reply": "2026-03-20T00:23:23.324557Z"
     }
    },
    "outputs": [],
@@ -272,10 +272,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:41:02.857244Z",
-     "iopub.status.busy": "2026-03-19T21:41:02.857143Z",
-     "iopub.status.idle": "2026-03-19T21:41:03.967600Z",
-     "shell.execute_reply": "2026-03-19T21:41:03.966691Z"
+     "iopub.execute_input": "2026-03-20T00:23:23.326206Z",
+     "iopub.status.busy": "2026-03-20T00:23:23.326113Z",
+     "iopub.status.idle": "2026-03-20T00:23:24.439839Z",
+     "shell.execute_reply": "2026-03-20T00:23:24.439278Z"
     }
    },
    "outputs": [],
diff --git a/nbs/image_points_transforms.ipynb b/nbs/image_points_transforms.ipynb
index 14c0f67..7d920d5 100644
--- a/nbs/image_points_transforms.ipynb
+++ b/nbs/image_points_transforms.ipynb
@@ -52,10 +52,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:41:06.603758Z",
-     "iopub.status.busy": "2026-03-19T21:41:06.603401Z",
-     "iopub.status.idle": "2026-03-19T21:41:07.237142Z",
-     "shell.execute_reply": "2026-03-19T21:41:07.235806Z"
+     "iopub.execute_input": "2026-03-20T00:23:26.954758Z",
+     "iopub.status.busy": "2026-03-20T00:23:26.954639Z",
+     "iopub.status.idle": "2026-03-20T00:23:27.182916Z",
+     "shell.execute_reply": "2026-03-20T00:23:27.182578Z"
     }
    },
    "outputs": [],
@@ -84,10 +84,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:41:07.269323Z",
-     "iopub.status.busy": "2026-03-19T21:41:07.269012Z",
-     "iopub.status.idle": "2026-03-19T21:41:08.438887Z",
-     "shell.execute_reply": "2026-03-19T21:41:08.438529Z"
+     "iopub.execute_input": "2026-03-20T00:23:27.204967Z",
+     "iopub.status.busy": "2026-03-20T00:23:27.204749Z",
+     "iopub.status.idle": "2026-03-20T00:23:28.239996Z",
+     "shell.execute_reply": "2026-03-20T00:23:28.239633Z"
     }
    },
    "outputs": [],
@@ -104,10 +104,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:41:08.440847Z",
-     "iopub.status.busy": "2026-03-19T21:41:08.440669Z",
-     "iopub.status.idle": "2026-03-19T21:41:08.446477Z",
-     "shell.execute_reply": "2026-03-19T21:41:08.446180Z"
+     "iopub.execute_input": "2026-03-20T00:23:28.242155Z",
+     "iopub.status.busy": "2026-03-20T00:23:28.241743Z",
+     "iopub.status.idle": "2026-03-20T00:23:28.247806Z",
+     "shell.execute_reply": "2026-03-20T00:23:28.247559Z"
     }
    },
    "outputs": [],
@@ -129,10 +129,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:41:08.447908Z",
-     "iopub.status.busy": "2026-03-19T21:41:08.447821Z",
-     "iopub.status.idle": "2026-03-19T21:41:08.512583Z",
-     "shell.execute_reply": "2026-03-19T21:41:08.511492Z"
+     "iopub.execute_input": "2026-03-20T00:23:28.249228Z",
+     "iopub.status.busy": "2026-03-20T00:23:28.249138Z",
+     "iopub.status.idle": "2026-03-20T00:23:28.312002Z",
+     "shell.execute_reply": "2026-03-20T00:23:28.311721Z"
     }
    },
    "outputs": [],
@@ -159,10 +159,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:41:08.514873Z",
-     "iopub.status.busy": "2026-03-19T21:41:08.514729Z",
-     "iopub.status.idle": "2026-03-19T21:41:08.617348Z",
-     "shell.execute_reply": "2026-03-19T21:41:08.617088Z"
+     "iopub.execute_input": "2026-03-20T00:23:28.313481Z",
+     "iopub.status.busy": "2026-03-20T00:23:28.313368Z",
+     "iopub.status.idle": "2026-03-20T00:23:28.414034Z",
+     "shell.execute_reply": "2026-03-20T00:23:28.413735Z"
     }
    },
    "outputs": [],
@@ -196,10 +196,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:41:08.618827Z",
-     "iopub.status.busy": "2026-03-19T21:41:08.618708Z",
-     "iopub.status.idle": "2026-03-19T21:41:08.675334Z",
-     "shell.execute_reply": "2026-03-19T21:41:08.675057Z"
+     "iopub.execute_input": "2026-03-20T00:23:28.415622Z",
+     "iopub.status.busy": "2026-03-20T00:23:28.415495Z",
+     "iopub.status.idle": "2026-03-20T00:23:28.470746Z",
+     "shell.execute_reply": "2026-03-20T00:23:28.470476Z"
     }
    },
    "outputs": [],
diff --git a/nbs/image_stitching.ipynb b/nbs/image_stitching.ipynb
index 54135fe..79aeb7c 100644
--- a/nbs/image_stitching.ipynb
+++ b/nbs/image_stitching.ipynb
@@ -61,7 +61,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:26:37.372795Z",
+     "iopub.status.busy": "2026-03-20T00:26:37.371036Z",
+     "iopub.status.idle": "2026-03-20T00:26:38.503281Z",
+     "shell.execute_reply": "2026-03-20T00:26:38.501871Z"
+    }
+   },
    "outputs": [],
    "source": [
     "import io\n",
@@ -87,7 +94,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:26:38.530699Z",
+     "iopub.status.busy": "2026-03-20T00:26:38.530497Z",
+     "iopub.status.idle": "2026-03-20T00:26:39.604965Z",
+     "shell.execute_reply": "2026-03-20T00:26:39.604573Z"
+    }
+   },
    "outputs": [],
    "source": [
     "import kornia as K\n",
@@ -95,13 +109,11 @@
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import torch\n",
-    "\n",
-    "%%capture\n",
     "from PIL import Image\n",
     "\n",
     "\n",
     "def load_images(fnames):\n",
-    "    return [K.image_to_tensor(np.array(Image.open(fn).convert(\"RGB\")))[None, ...] for fn in fnames]\n",
+    "    return [K.image_to_tensor(np.array(Image.open(fn).convert(\"RGB\"))).float()[None, ...] / 255.0 for fn in fnames]\n",
     "\n",
     "\n",
     "imgs = load_images([\"prtn00.jpg\", \"prtn01.jpg\"])"
@@ -117,7 +129,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:26:39.606901Z",
+     "iopub.status.busy": "2026-03-20T00:26:39.606715Z",
+     "iopub.status.idle": "2026-03-20T00:26:41.597345Z",
+     "shell.execute_reply": "2026-03-20T00:26:41.596873Z"
+    }
+   },
    "outputs": [],
    "source": [
     "from kornia.contrib import ImageStitcher\n",
@@ -141,7 +160,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:26:41.599273Z",
+     "iopub.status.busy": "2026-03-20T00:26:41.599126Z",
+     "iopub.status.idle": "2026-03-20T00:26:43.987665Z",
+     "shell.execute_reply": "2026-03-20T00:26:43.985978Z"
+    }
+   },
    "outputs": [],
    "source": [
     "download_image(\"https://github.com/daeyun/Image-Stitching/blob/master/img/hill/1.JPG?raw=true\")\n",
@@ -152,7 +178,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:26:43.993876Z",
+     "iopub.status.busy": "2026-03-20T00:26:43.992845Z",
+     "iopub.status.idle": "2026-03-20T00:26:44.018661Z",
+     "shell.execute_reply": "2026-03-20T00:26:44.017942Z"
+    }
+   },
    "outputs": [],
    "source": [
     "imgs = load_images([\"1.JPG\", \"2.JPG\", \"3.JPG\"])"
@@ -161,7 +194,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:26:44.021089Z",
+     "iopub.status.busy": "2026-03-20T00:26:44.020932Z",
+     "iopub.status.idle": "2026-03-20T00:26:44.180522Z",
+     "shell.execute_reply": "2026-03-20T00:26:44.180199Z"
+    }
+   },
    "outputs": [],
    "source": [
     "f, axarr = plt.subplots(1, 3, figsize=(16, 6))\n",
@@ -177,7 +217,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:26:44.183805Z",
+     "iopub.status.busy": "2026-03-20T00:26:44.183678Z",
+     "iopub.status.idle": "2026-03-20T00:26:45.875541Z",
+     "shell.execute_reply": "2026-03-20T00:26:45.875203Z"
+    }
+   },
    "outputs": [],
    "source": [
     "matcher = KF.LocalFeatureMatcher(KF.GFTTAffNetHardNet(100), KF.DescriptorMatcher(\"snn\", 0.8))\n",
@@ -190,7 +237,14 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "execution": {
+     "iopub.execute_input": "2026-03-20T00:26:45.877247Z",
+     "iopub.status.busy": "2026-03-20T00:26:45.877136Z",
+     "iopub.status.idle": "2026-03-20T00:26:46.010761Z",
+     "shell.execute_reply": "2026-03-20T00:26:46.010444Z"
+    }
+   },
    "outputs": [],
    "source": [
     "plt.figure(figsize=(16, 16))\n",
@@ -214,7 +268,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.10"
+   "version": "3.10.15"
   }
  },
  "nbformat": 4,
diff --git a/nbs/line_detection_and_matching_sold2.ipynb b/nbs/line_detection_and_matching_sold2.ipynb
index 19e388c..9486bf1 100644
--- a/nbs/line_detection_and_matching_sold2.ipynb
+++ b/nbs/line_detection_and_matching_sold2.ipynb
@@ -62,10 +62,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:17.833230Z",
-     "iopub.status.busy": "2026-03-19T21:43:17.832750Z",
-     "iopub.status.idle": "2026-03-19T21:43:18.963897Z",
-     "shell.execute_reply": "2026-03-19T21:43:18.962787Z"
+     "iopub.execute_input": "2026-03-20T00:26:48.464110Z",
+     "iopub.status.busy": "2026-03-20T00:26:48.463186Z",
+     "iopub.status.idle": "2026-03-20T00:26:49.749948Z",
+     "shell.execute_reply": "2026-03-20T00:26:49.748917Z"
     }
    },
    "outputs": [],
@@ -102,10 +102,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:18.987360Z",
-     "iopub.status.busy": "2026-03-19T21:43:18.987172Z",
-     "iopub.status.idle": "2026-03-19T21:43:19.833262Z",
-     "shell.execute_reply": "2026-03-19T21:43:19.832917Z"
+     "iopub.execute_input": "2026-03-20T00:26:49.782917Z",
+     "iopub.status.busy": "2026-03-20T00:26:49.782717Z",
+     "iopub.status.idle": "2026-03-20T00:26:50.592173Z",
+     "shell.execute_reply": "2026-03-20T00:26:50.591881Z"
     }
    },
    "outputs": [],
@@ -129,10 +129,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:19.835050Z",
-     "iopub.status.busy": "2026-03-19T21:43:19.834883Z",
-     "iopub.status.idle": "2026-03-19T21:43:19.850454Z",
-     "shell.execute_reply": "2026-03-19T21:43:19.850171Z"
+     "iopub.execute_input": "2026-03-20T00:26:50.593902Z",
+     "iopub.status.busy": "2026-03-20T00:26:50.593776Z",
+     "iopub.status.idle": "2026-03-20T00:26:50.607405Z",
+     "shell.execute_reply": "2026-03-20T00:26:50.607132Z"
     }
    },
    "outputs": [],
@@ -160,10 +160,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:19.852092Z",
-     "iopub.status.busy": "2026-03-19T21:43:19.851965Z",
-     "iopub.status.idle": "2026-03-19T21:43:19.856046Z",
-     "shell.execute_reply": "2026-03-19T21:43:19.855713Z"
+     "iopub.execute_input": "2026-03-20T00:26:50.608808Z",
+     "iopub.status.busy": "2026-03-20T00:26:50.608712Z",
+     "iopub.status.idle": "2026-03-20T00:26:50.612654Z",
+     "shell.execute_reply": "2026-03-20T00:26:50.612377Z"
     }
    },
    "outputs": [],
@@ -178,10 +178,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:19.857538Z",
-     "iopub.status.busy": "2026-03-19T21:43:19.857441Z",
-     "iopub.status.idle": "2026-03-19T21:43:19.859890Z",
-     "shell.execute_reply": "2026-03-19T21:43:19.859642Z"
+     "iopub.execute_input": "2026-03-20T00:26:50.614161Z",
+     "iopub.status.busy": "2026-03-20T00:26:50.614066Z",
+     "iopub.status.idle": "2026-03-20T00:26:50.616204Z",
+     "shell.execute_reply": "2026-03-20T00:26:50.615967Z"
     }
    },
    "outputs": [],
@@ -194,10 +194,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:19.861327Z",
-     "iopub.status.busy": "2026-03-19T21:43:19.861188Z",
-     "iopub.status.idle": "2026-03-19T21:43:19.864452Z",
-     "shell.execute_reply": "2026-03-19T21:43:19.863905Z"
+     "iopub.execute_input": "2026-03-20T00:26:50.617459Z",
+     "iopub.status.busy": "2026-03-20T00:26:50.617375Z",
+     "iopub.status.idle": "2026-03-20T00:26:50.620106Z",
+     "shell.execute_reply": "2026-03-20T00:26:50.619760Z"
     }
    },
    "outputs": [],
@@ -227,10 +227,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:19.866124Z",
-     "iopub.status.busy": "2026-03-19T21:43:19.865996Z",
-     "iopub.status.idle": "2026-03-19T21:43:19.988843Z",
-     "shell.execute_reply": "2026-03-19T21:43:19.988541Z"
+     "iopub.execute_input": "2026-03-20T00:26:50.621541Z",
+     "iopub.status.busy": "2026-03-20T00:26:50.621448Z",
+     "iopub.status.idle": "2026-03-20T00:26:50.737103Z",
+     "shell.execute_reply": "2026-03-20T00:26:50.736794Z"
     }
    },
    "outputs": [],
@@ -251,10 +251,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:19.990520Z",
-     "iopub.status.busy": "2026-03-19T21:43:19.990404Z",
-     "iopub.status.idle": "2026-03-19T21:43:28.210543Z",
-     "shell.execute_reply": "2026-03-19T21:43:28.210237Z"
+     "iopub.execute_input": "2026-03-20T00:26:50.738735Z",
+     "iopub.status.busy": "2026-03-20T00:26:50.738649Z",
+     "iopub.status.idle": "2026-03-20T00:26:59.539273Z",
+     "shell.execute_reply": "2026-03-20T00:26:59.538903Z"
     }
    },
    "outputs": [],
@@ -278,10 +278,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:28.212446Z",
-     "iopub.status.busy": "2026-03-19T21:43:28.212286Z",
-     "iopub.status.idle": "2026-03-19T21:43:28.214661Z",
-     "shell.execute_reply": "2026-03-19T21:43:28.214395Z"
+     "iopub.execute_input": "2026-03-20T00:26:59.541175Z",
+     "iopub.status.busy": "2026-03-20T00:26:59.541038Z",
+     "iopub.status.idle": "2026-03-20T00:26:59.543245Z",
+     "shell.execute_reply": "2026-03-20T00:26:59.543012Z"
     }
    },
    "outputs": [],
@@ -294,10 +294,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:28.215952Z",
-     "iopub.status.busy": "2026-03-19T21:43:28.215845Z",
-     "iopub.status.idle": "2026-03-19T21:43:28.217588Z",
-     "shell.execute_reply": "2026-03-19T21:43:28.217352Z"
+     "iopub.execute_input": "2026-03-20T00:26:59.544470Z",
+     "iopub.status.busy": "2026-03-20T00:26:59.544367Z",
+     "iopub.status.idle": "2026-03-20T00:26:59.546166Z",
+     "shell.execute_reply": "2026-03-20T00:26:59.545932Z"
     }
    },
    "outputs": [],
@@ -320,10 +320,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:28.218915Z",
-     "iopub.status.busy": "2026-03-19T21:43:28.218830Z",
-     "iopub.status.idle": "2026-03-19T21:43:28.265703Z",
-     "shell.execute_reply": "2026-03-19T21:43:28.265237Z"
+     "iopub.execute_input": "2026-03-20T00:26:59.547501Z",
+     "iopub.status.busy": "2026-03-20T00:26:59.547396Z",
+     "iopub.status.idle": "2026-03-20T00:26:59.584640Z",
+     "shell.execute_reply": "2026-03-20T00:26:59.584274Z"
     }
    },
    "outputs": [],
@@ -337,10 +337,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:28.267746Z",
-     "iopub.status.busy": "2026-03-19T21:43:28.267599Z",
-     "iopub.status.idle": "2026-03-19T21:43:28.269818Z",
-     "shell.execute_reply": "2026-03-19T21:43:28.269555Z"
+     "iopub.execute_input": "2026-03-20T00:26:59.586574Z",
+     "iopub.status.busy": "2026-03-20T00:26:59.586299Z",
+     "iopub.status.idle": "2026-03-20T00:26:59.588847Z",
+     "shell.execute_reply": "2026-03-20T00:26:59.588512Z"
     }
    },
    "outputs": [],
@@ -366,10 +366,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:28.271359Z",
-     "iopub.status.busy": "2026-03-19T21:43:28.271237Z",
-     "iopub.status.idle": "2026-03-19T21:43:28.478708Z",
-     "shell.execute_reply": "2026-03-19T21:43:28.478389Z"
+     "iopub.execute_input": "2026-03-20T00:26:59.590544Z",
+     "iopub.status.busy": "2026-03-20T00:26:59.590422Z",
+     "iopub.status.idle": "2026-03-20T00:26:59.831394Z",
+     "shell.execute_reply": "2026-03-20T00:26:59.831095Z"
     }
    },
    "outputs": [],
@@ -481,10 +481,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:28.480700Z",
-     "iopub.status.busy": "2026-03-19T21:43:28.480496Z",
-     "iopub.status.idle": "2026-03-19T21:43:28.822664Z",
-     "shell.execute_reply": "2026-03-19T21:43:28.822344Z"
+     "iopub.execute_input": "2026-03-20T00:26:59.832986Z",
+     "iopub.status.busy": "2026-03-20T00:26:59.832839Z",
+     "iopub.status.idle": "2026-03-20T00:27:00.192648Z",
+     "shell.execute_reply": "2026-03-20T00:27:00.192270Z"
     }
    },
    "outputs": [],
@@ -501,10 +501,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:28.826425Z",
-     "iopub.status.busy": "2026-03-19T21:43:28.826254Z",
-     "iopub.status.idle": "2026-03-19T21:43:29.181048Z",
-     "shell.execute_reply": "2026-03-19T21:43:29.180742Z"
+     "iopub.execute_input": "2026-03-20T00:27:00.196613Z",
+     "iopub.status.busy": "2026-03-20T00:27:00.196427Z",
+     "iopub.status.idle": "2026-03-20T00:27:00.550066Z",
+     "shell.execute_reply": "2026-03-20T00:27:00.549398Z"
     }
    },
    "outputs": [],
@@ -534,10 +534,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:29.185077Z",
-     "iopub.status.busy": "2026-03-19T21:43:29.184847Z",
-     "iopub.status.idle": "2026-03-19T21:43:29.186851Z",
-     "shell.execute_reply": "2026-03-19T21:43:29.186613Z"
+     "iopub.execute_input": "2026-03-20T00:27:00.553849Z",
+     "iopub.status.busy": "2026-03-20T00:27:00.553670Z",
+     "iopub.status.idle": "2026-03-20T00:27:00.556242Z",
+     "shell.execute_reply": "2026-03-20T00:27:00.555615Z"
     }
    },
    "outputs": [],
@@ -557,10 +557,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:29.188107Z",
-     "iopub.status.busy": "2026-03-19T21:43:29.188010Z",
-     "iopub.status.idle": "2026-03-19T21:43:29.611412Z",
-     "shell.execute_reply": "2026-03-19T21:43:29.611090Z"
+     "iopub.execute_input": "2026-03-20T00:27:00.558188Z",
+     "iopub.status.busy": "2026-03-20T00:27:00.558050Z",
+     "iopub.status.idle": "2026-03-20T00:27:01.001310Z",
+     "shell.execute_reply": "2026-03-20T00:27:01.000944Z"
     }
    },
    "outputs": [],
@@ -580,10 +580,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:29.613325Z",
-     "iopub.status.busy": "2026-03-19T21:43:29.613168Z",
-     "iopub.status.idle": "2026-03-19T21:43:29.628617Z",
-     "shell.execute_reply": "2026-03-19T21:43:29.628251Z"
+     "iopub.execute_input": "2026-03-20T00:27:01.003428Z",
+     "iopub.status.busy": "2026-03-20T00:27:01.003272Z",
+     "iopub.status.idle": "2026-03-20T00:27:01.018338Z",
+     "shell.execute_reply": "2026-03-20T00:27:01.018017Z"
     }
    },
    "outputs": [],
@@ -603,10 +603,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:29.630653Z",
-     "iopub.status.busy": "2026-03-19T21:43:29.630473Z",
-     "iopub.status.idle": "2026-03-19T21:43:29.773721Z",
-     "shell.execute_reply": "2026-03-19T21:43:29.773435Z"
+     "iopub.execute_input": "2026-03-20T00:27:01.020202Z",
+     "iopub.status.busy": "2026-03-20T00:27:01.020017Z",
+     "iopub.status.idle": "2026-03-20T00:27:01.166223Z",
+     "shell.execute_reply": "2026-03-20T00:27:01.165902Z"
     }
    },
    "outputs": [],
@@ -623,10 +623,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:29.777069Z",
-     "iopub.status.busy": "2026-03-19T21:43:29.776949Z",
-     "iopub.status.idle": "2026-03-19T21:43:29.891024Z",
-     "shell.execute_reply": "2026-03-19T21:43:29.890718Z"
+     "iopub.execute_input": "2026-03-20T00:27:01.170075Z",
+     "iopub.status.busy": "2026-03-20T00:27:01.169926Z",
+     "iopub.status.idle": "2026-03-20T00:27:01.283467Z",
+     "shell.execute_reply": "2026-03-20T00:27:01.283173Z"
     }
    },
    "outputs": [],
diff --git a/nbs/morphology_101.ipynb b/nbs/morphology_101.ipynb
index b768f9a..c103b69 100644
--- a/nbs/morphology_101.ipynb
+++ b/nbs/morphology_101.ipynb
@@ -82,10 +82,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:32.209439Z",
-     "iopub.status.busy": "2026-03-19T21:43:32.208897Z",
-     "iopub.status.idle": "2026-03-19T21:43:33.390416Z",
-     "shell.execute_reply": "2026-03-19T21:43:33.390102Z"
+     "iopub.execute_input": "2026-03-20T00:27:03.695341Z",
+     "iopub.status.busy": "2026-03-20T00:27:03.694676Z",
+     "iopub.status.idle": "2026-03-20T00:27:03.931419Z",
+     "shell.execute_reply": "2026-03-20T00:27:03.931042Z"
     }
    },
    "outputs": [],
@@ -125,10 +125,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:33.413608Z",
-     "iopub.status.busy": "2026-03-19T21:43:33.413411Z",
-     "iopub.status.idle": "2026-03-19T21:43:34.231229Z",
-     "shell.execute_reply": "2026-03-19T21:43:34.230908Z"
+     "iopub.execute_input": "2026-03-20T00:27:03.954908Z",
+     "iopub.status.busy": "2026-03-20T00:27:03.954720Z",
+     "iopub.status.idle": "2026-03-20T00:27:04.934283Z",
+     "shell.execute_reply": "2026-03-20T00:27:04.933987Z"
     }
    },
    "outputs": [],
@@ -146,10 +146,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:34.232800Z",
-     "iopub.status.busy": "2026-03-19T21:43:34.232667Z",
-     "iopub.status.idle": "2026-03-19T21:43:34.240691Z",
-     "shell.execute_reply": "2026-03-19T21:43:34.240412Z"
+     "iopub.execute_input": "2026-03-20T00:27:04.935927Z",
+     "iopub.status.busy": "2026-03-20T00:27:04.935777Z",
+     "iopub.status.idle": "2026-03-20T00:27:04.945394Z",
+     "shell.execute_reply": "2026-03-20T00:27:04.945122Z"
     }
    },
    "outputs": [],
@@ -178,10 +178,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:34.242281Z",
-     "iopub.status.busy": "2026-03-19T21:43:34.242189Z",
-     "iopub.status.idle": "2026-03-19T21:43:34.244191Z",
-     "shell.execute_reply": "2026-03-19T21:43:34.243938Z"
+     "iopub.execute_input": "2026-03-20T00:27:04.946921Z",
+     "iopub.status.busy": "2026-03-20T00:27:04.946819Z",
+     "iopub.status.idle": "2026-03-20T00:27:04.948652Z",
+     "shell.execute_reply": "2026-03-20T00:27:04.948431Z"
     }
    },
    "outputs": [],
@@ -210,10 +210,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:34.245834Z",
-     "iopub.status.busy": "2026-03-19T21:43:34.245732Z",
-     "iopub.status.idle": "2026-03-19T21:43:34.447859Z",
-     "shell.execute_reply": "2026-03-19T21:43:34.447376Z"
+     "iopub.execute_input": "2026-03-20T00:27:04.949943Z",
+     "iopub.status.busy": "2026-03-20T00:27:04.949846Z",
+     "iopub.status.idle": "2026-03-20T00:27:05.146622Z",
+     "shell.execute_reply": "2026-03-20T00:27:05.146302Z"
     }
    },
    "outputs": [],
@@ -270,10 +270,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:34.449706Z",
-     "iopub.status.busy": "2026-03-19T21:43:34.449539Z",
-     "iopub.status.idle": "2026-03-19T21:43:34.550689Z",
-     "shell.execute_reply": "2026-03-19T21:43:34.550387Z"
+     "iopub.execute_input": "2026-03-20T00:27:05.148516Z",
+     "iopub.status.busy": "2026-03-20T00:27:05.148357Z",
+     "iopub.status.idle": "2026-03-20T00:27:05.246999Z",
+     "shell.execute_reply": "2026-03-20T00:27:05.246676Z"
     }
    },
    "outputs": [],
@@ -296,10 +296,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:34.554319Z",
-     "iopub.status.busy": "2026-03-19T21:43:34.554197Z",
-     "iopub.status.idle": "2026-03-19T21:43:34.645971Z",
-     "shell.execute_reply": "2026-03-19T21:43:34.645621Z"
+     "iopub.execute_input": "2026-03-20T00:27:05.250153Z",
+     "iopub.status.busy": "2026-03-20T00:27:05.250042Z",
+     "iopub.status.idle": "2026-03-20T00:27:05.341953Z",
+     "shell.execute_reply": "2026-03-20T00:27:05.341650Z"
     }
    },
    "outputs": [],
@@ -320,10 +320,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:34.649658Z",
-     "iopub.status.busy": "2026-03-19T21:43:34.649524Z",
-     "iopub.status.idle": "2026-03-19T21:43:34.743246Z",
-     "shell.execute_reply": "2026-03-19T21:43:34.742915Z"
+     "iopub.execute_input": "2026-03-20T00:27:05.345403Z",
+     "iopub.status.busy": "2026-03-20T00:27:05.345286Z",
+     "iopub.status.idle": "2026-03-20T00:27:05.437381Z",
+     "shell.execute_reply": "2026-03-20T00:27:05.437063Z"
     }
    },
    "outputs": [],
@@ -344,10 +344,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:34.747070Z",
-     "iopub.status.busy": "2026-03-19T21:43:34.746911Z",
-     "iopub.status.idle": "2026-03-19T21:43:34.845499Z",
-     "shell.execute_reply": "2026-03-19T21:43:34.845100Z"
+     "iopub.execute_input": "2026-03-20T00:27:05.440488Z",
+     "iopub.status.busy": "2026-03-20T00:27:05.440370Z",
+     "iopub.status.idle": "2026-03-20T00:27:05.534641Z",
+     "shell.execute_reply": "2026-03-20T00:27:05.534332Z"
     }
    },
    "outputs": [],
@@ -368,10 +368,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:34.848712Z",
-     "iopub.status.busy": "2026-03-19T21:43:34.848554Z",
-     "iopub.status.idle": "2026-03-19T21:43:34.945326Z",
-     "shell.execute_reply": "2026-03-19T21:43:34.944956Z"
+     "iopub.execute_input": "2026-03-20T00:27:05.537473Z",
+     "iopub.status.busy": "2026-03-20T00:27:05.537365Z",
+     "iopub.status.idle": "2026-03-20T00:27:05.629892Z",
+     "shell.execute_reply": "2026-03-20T00:27:05.629596Z"
     }
    },
    "outputs": [],
@@ -392,10 +392,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:34.948944Z",
-     "iopub.status.busy": "2026-03-19T21:43:34.948791Z",
-     "iopub.status.idle": "2026-03-19T21:43:35.044851Z",
-     "shell.execute_reply": "2026-03-19T21:43:35.044341Z"
+     "iopub.execute_input": "2026-03-20T00:27:05.632960Z",
+     "iopub.status.busy": "2026-03-20T00:27:05.632849Z",
+     "iopub.status.idle": "2026-03-20T00:27:05.729785Z",
+     "shell.execute_reply": "2026-03-20T00:27:05.729480Z"
     }
    },
    "outputs": [],
@@ -416,10 +416,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:35.048328Z",
-     "iopub.status.busy": "2026-03-19T21:43:35.048158Z",
-     "iopub.status.idle": "2026-03-19T21:43:35.145246Z",
-     "shell.execute_reply": "2026-03-19T21:43:35.144901Z"
+     "iopub.execute_input": "2026-03-20T00:27:05.732754Z",
+     "iopub.status.busy": "2026-03-20T00:27:05.732638Z",
+     "iopub.status.idle": "2026-03-20T00:27:05.829059Z",
+     "shell.execute_reply": "2026-03-20T00:27:05.828603Z"
     }
    },
    "outputs": [],
diff --git a/nbs/resize_antialias.ipynb b/nbs/resize_antialias.ipynb
index 3025d16..5a92356 100644
--- a/nbs/resize_antialias.ipynb
+++ b/nbs/resize_antialias.ipynb
@@ -33,10 +33,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:28:24.639861Z",
-     "iopub.status.busy": "2026-03-19T21:28:24.639230Z",
-     "iopub.status.idle": "2026-03-19T21:28:27.388446Z",
-     "shell.execute_reply": "2026-03-19T21:28:27.387951Z"
+     "iopub.execute_input": "2026-03-20T00:27:08.046272Z",
+     "iopub.status.busy": "2026-03-20T00:27:08.045798Z",
+     "iopub.status.idle": "2026-03-20T00:27:10.948488Z",
+     "shell.execute_reply": "2026-03-20T00:27:10.947878Z"
     }
    },
    "outputs": [],
@@ -60,10 +60,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:28:27.391068Z",
-     "iopub.status.busy": "2026-03-19T21:28:27.390897Z",
-     "iopub.status.idle": "2026-03-19T21:28:27.921529Z",
-     "shell.execute_reply": "2026-03-19T21:28:27.920878Z"
+     "iopub.execute_input": "2026-03-20T00:27:10.950725Z",
+     "iopub.status.busy": "2026-03-20T00:27:10.950506Z",
+     "iopub.status.idle": "2026-03-20T00:27:11.549468Z",
+     "shell.execute_reply": "2026-03-20T00:27:11.548854Z"
     }
    },
    "outputs": [],
@@ -93,10 +93,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:28:27.945910Z",
-     "iopub.status.busy": "2026-03-19T21:28:27.945754Z",
-     "iopub.status.idle": "2026-03-19T21:28:28.975783Z",
-     "shell.execute_reply": "2026-03-19T21:28:28.975441Z"
+     "iopub.execute_input": "2026-03-20T00:27:11.576787Z",
+     "iopub.status.busy": "2026-03-20T00:27:11.576465Z",
+     "iopub.status.idle": "2026-03-20T00:27:12.574920Z",
+     "shell.execute_reply": "2026-03-20T00:27:12.574576Z"
     }
    },
    "outputs": [],
@@ -114,10 +114,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:28:28.978239Z",
-     "iopub.status.busy": "2026-03-19T21:28:28.977959Z",
-     "iopub.status.idle": "2026-03-19T21:28:29.057538Z",
-     "shell.execute_reply": "2026-03-19T21:28:29.057208Z"
+     "iopub.execute_input": "2026-03-20T00:27:12.576508Z",
+     "iopub.status.busy": "2026-03-20T00:27:12.576378Z",
+     "iopub.status.idle": "2026-03-20T00:27:12.658836Z",
+     "shell.execute_reply": "2026-03-20T00:27:12.658558Z"
     }
    },
    "outputs": [],
@@ -131,7 +131,7 @@
     "        ax.axis(\"off\")\n",
     "\n",
     "\n",
-    "data: torch.Tensor = K.image_to_tensor(np.array(Image.open(\"drslump.jpg\").convert(\"RGB\")))[None, ...]  # BxCxHxW\n",
+    "data: torch.Tensor = K.image_to_tensor(np.array(Image.open(\"drslump.jpg\").convert(\"RGB\"))).float()[None, ...] / 255.0  # BxCxHxW\n",
     "\n",
     "# plot\n",
     "imshow(data)"
@@ -149,10 +149,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:28:29.061702Z",
-     "iopub.status.busy": "2026-03-19T21:28:29.061582Z",
-     "iopub.status.idle": "2026-03-19T21:28:29.136593Z",
-     "shell.execute_reply": "2026-03-19T21:28:29.136271Z"
+     "iopub.execute_input": "2026-03-20T00:27:12.662716Z",
+     "iopub.status.busy": "2026-03-20T00:27:12.662600Z",
+     "iopub.status.idle": "2026-03-20T00:27:12.760292Z",
+     "shell.execute_reply": "2026-03-20T00:27:12.760010Z"
     }
    },
    "outputs": [],
diff --git a/nbs/rotate_affine.ipynb b/nbs/rotate_affine.ipynb
index ad07ea6..b461156 100644
--- a/nbs/rotate_affine.ipynb
+++ b/nbs/rotate_affine.ipynb
@@ -41,10 +41,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:37.297791Z",
-     "iopub.status.busy": "2026-03-19T21:43:37.296747Z",
-     "iopub.status.idle": "2026-03-19T21:43:38.489329Z",
-     "shell.execute_reply": "2026-03-19T21:43:38.488409Z"
+     "iopub.execute_input": "2026-03-20T00:27:14.948818Z",
+     "iopub.status.busy": "2026-03-20T00:27:14.948190Z",
+     "iopub.status.idle": "2026-03-20T00:27:15.513172Z",
+     "shell.execute_reply": "2026-03-20T00:27:15.512457Z"
     }
    },
    "outputs": [],
@@ -74,10 +74,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:38.523757Z",
-     "iopub.status.busy": "2026-03-19T21:43:38.523580Z",
-     "iopub.status.idle": "2026-03-19T21:43:40.300844Z",
-     "shell.execute_reply": "2026-03-19T21:43:40.300538Z"
+     "iopub.execute_input": "2026-03-20T00:27:15.536189Z",
+     "iopub.status.busy": "2026-03-20T00:27:15.536013Z",
+     "iopub.status.idle": "2026-03-20T00:27:17.347379Z",
+     "shell.execute_reply": "2026-03-20T00:27:17.347059Z"
     }
    },
    "outputs": [],
@@ -105,10 +105,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:40.302453Z",
-     "iopub.status.busy": "2026-03-19T21:43:40.302311Z",
-     "iopub.status.idle": "2026-03-19T21:43:40.308663Z",
-     "shell.execute_reply": "2026-03-19T21:43:40.308392Z"
+     "iopub.execute_input": "2026-03-20T00:27:17.349088Z",
+     "iopub.status.busy": "2026-03-20T00:27:17.348954Z",
+     "iopub.status.idle": "2026-03-20T00:27:17.355763Z",
+     "shell.execute_reply": "2026-03-20T00:27:17.355396Z"
     }
    },
    "outputs": [],
@@ -121,10 +121,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:40.309984Z",
-     "iopub.status.busy": "2026-03-19T21:43:40.309886Z",
-     "iopub.status.idle": "2026-03-19T21:43:40.312060Z",
-     "shell.execute_reply": "2026-03-19T21:43:40.311803Z"
+     "iopub.execute_input": "2026-03-20T00:27:17.357505Z",
+     "iopub.status.busy": "2026-03-20T00:27:17.357400Z",
+     "iopub.status.idle": "2026-03-20T00:27:17.359599Z",
+     "shell.execute_reply": "2026-03-20T00:27:17.359332Z"
     }
    },
    "outputs": [],
@@ -143,10 +143,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:40.313422Z",
-     "iopub.status.busy": "2026-03-19T21:43:40.313327Z",
-     "iopub.status.idle": "2026-03-19T21:43:40.348807Z",
-     "shell.execute_reply": "2026-03-19T21:43:40.348367Z"
+     "iopub.execute_input": "2026-03-20T00:27:17.360934Z",
+     "iopub.status.busy": "2026-03-20T00:27:17.360828Z",
+     "iopub.status.idle": "2026-03-20T00:27:17.396327Z",
+     "shell.execute_reply": "2026-03-20T00:27:17.396032Z"
     }
    },
    "outputs": [],
@@ -166,10 +166,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:40.350387Z",
-     "iopub.status.busy": "2026-03-19T21:43:40.350282Z",
-     "iopub.status.idle": "2026-03-19T21:43:40.353279Z",
-     "shell.execute_reply": "2026-03-19T21:43:40.353018Z"
+     "iopub.execute_input": "2026-03-20T00:27:17.397976Z",
+     "iopub.status.busy": "2026-03-20T00:27:17.397862Z",
+     "iopub.status.idle": "2026-03-20T00:27:17.400938Z",
+     "shell.execute_reply": "2026-03-20T00:27:17.400643Z"
     }
    },
    "outputs": [],
@@ -202,10 +202,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:40.354599Z",
-     "iopub.status.busy": "2026-03-19T21:43:40.354513Z",
-     "iopub.status.idle": "2026-03-19T21:43:40.383802Z",
-     "shell.execute_reply": "2026-03-19T21:43:40.383515Z"
+     "iopub.execute_input": "2026-03-20T00:27:17.402388Z",
+     "iopub.status.busy": "2026-03-20T00:27:17.402267Z",
+     "iopub.status.idle": "2026-03-20T00:27:17.431374Z",
+     "shell.execute_reply": "2026-03-20T00:27:17.431079Z"
     }
    },
    "outputs": [],
@@ -228,10 +228,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:40.385301Z",
-     "iopub.status.busy": "2026-03-19T21:43:40.385175Z",
-     "iopub.status.idle": "2026-03-19T21:43:40.553092Z",
-     "shell.execute_reply": "2026-03-19T21:43:40.552729Z"
+     "iopub.execute_input": "2026-03-20T00:27:17.432912Z",
+     "iopub.status.busy": "2026-03-20T00:27:17.432788Z",
+     "iopub.status.idle": "2026-03-20T00:27:17.597823Z",
+     "shell.execute_reply": "2026-03-20T00:27:17.597496Z"
     }
    },
    "outputs": [],
diff --git a/nbs/total_variation_denoising.ipynb b/nbs/total_variation_denoising.ipynb
index 758eefb..5f3251b 100644
--- a/nbs/total_variation_denoising.ipynb
+++ b/nbs/total_variation_denoising.ipynb
@@ -42,10 +42,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:42.854597Z",
-     "iopub.status.busy": "2026-03-19T21:43:42.854044Z",
-     "iopub.status.idle": "2026-03-19T21:43:43.718437Z",
-     "shell.execute_reply": "2026-03-19T21:43:43.718020Z"
+     "iopub.execute_input": "2026-03-20T00:27:19.840328Z",
+     "iopub.status.busy": "2026-03-20T00:27:19.839765Z",
+     "iopub.status.idle": "2026-03-20T00:27:20.441875Z",
+     "shell.execute_reply": "2026-03-20T00:27:20.439420Z"
     }
    },
    "outputs": [],
@@ -75,10 +75,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:43.743933Z",
-     "iopub.status.busy": "2026-03-19T21:43:43.743758Z",
-     "iopub.status.idle": "2026-03-19T21:43:45.272814Z",
-     "shell.execute_reply": "2026-03-19T21:43:45.272303Z"
+     "iopub.execute_input": "2026-03-20T00:27:20.466997Z",
+     "iopub.status.busy": "2026-03-20T00:27:20.466809Z",
+     "iopub.status.idle": "2026-03-20T00:27:22.008628Z",
+     "shell.execute_reply": "2026-03-20T00:27:22.008303Z"
     }
    },
    "outputs": [],
@@ -97,10 +97,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:45.274987Z",
-     "iopub.status.busy": "2026-03-19T21:43:45.274781Z",
-     "iopub.status.idle": "2026-03-19T21:43:45.277279Z",
-     "shell.execute_reply": "2026-03-19T21:43:45.276875Z"
+     "iopub.execute_input": "2026-03-20T00:27:22.010575Z",
+     "iopub.status.busy": "2026-03-20T00:27:22.010377Z",
+     "iopub.status.idle": "2026-03-20T00:27:22.012753Z",
+     "shell.execute_reply": "2026-03-20T00:27:22.012465Z"
     }
    },
    "outputs": [],
@@ -118,10 +118,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:45.279176Z",
-     "iopub.status.busy": "2026-03-19T21:43:45.279009Z",
-     "iopub.status.idle": "2026-03-19T21:43:45.315830Z",
-     "shell.execute_reply": "2026-03-19T21:43:45.315525Z"
+     "iopub.execute_input": "2026-03-20T00:27:22.014178Z",
+     "iopub.status.busy": "2026-03-20T00:27:22.014061Z",
+     "iopub.status.idle": "2026-03-20T00:27:22.051866Z",
+     "shell.execute_reply": "2026-03-20T00:27:22.051538Z"
     }
    },
    "outputs": [],
@@ -145,10 +145,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:45.318307Z",
-     "iopub.status.busy": "2026-03-19T21:43:45.318213Z",
-     "iopub.status.idle": "2026-03-19T21:43:45.321454Z",
-     "shell.execute_reply": "2026-03-19T21:43:45.321189Z"
+     "iopub.execute_input": "2026-03-20T00:27:22.054700Z",
+     "iopub.status.busy": "2026-03-20T00:27:22.054516Z",
+     "iopub.status.idle": "2026-03-20T00:27:22.058097Z",
+     "shell.execute_reply": "2026-03-20T00:27:22.057782Z"
     }
    },
    "outputs": [],
@@ -190,10 +190,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:45.322712Z",
-     "iopub.status.busy": "2026-03-19T21:43:45.322631Z",
-     "iopub.status.idle": "2026-03-19T21:43:46.314072Z",
-     "shell.execute_reply": "2026-03-19T21:43:46.313687Z"
+     "iopub.execute_input": "2026-03-20T00:27:22.059611Z",
+     "iopub.status.busy": "2026-03-20T00:27:22.059518Z",
+     "iopub.status.idle": "2026-03-20T00:27:23.159805Z",
+     "shell.execute_reply": "2026-03-20T00:27:23.159430Z"
     }
    },
    "outputs": [],
@@ -220,10 +220,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:43:46.315670Z",
-     "iopub.status.busy": "2026-03-19T21:43:46.315544Z",
-     "iopub.status.idle": "2026-03-19T21:43:46.435210Z",
-     "shell.execute_reply": "2026-03-19T21:43:46.434913Z"
+     "iopub.execute_input": "2026-03-20T00:27:23.161656Z",
+     "iopub.status.busy": "2026-03-20T00:27:23.161514Z",
+     "iopub.status.idle": "2026-03-20T00:27:23.280127Z",
+     "shell.execute_reply": "2026-03-20T00:27:23.279799Z"
     }
    },
    "outputs": [],
diff --git a/nbs/unsharp_mask.ipynb b/nbs/unsharp_mask.ipynb
index dff0c28..05a2489 100644
--- a/nbs/unsharp_mask.ipynb
+++ b/nbs/unsharp_mask.ipynb
@@ -53,10 +53,10 @@
    "id": "7ad15348",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:28:39.472301Z",
-     "iopub.status.busy": "2026-03-19T21:28:39.471806Z",
-     "iopub.status.idle": "2026-03-19T21:28:40.482195Z",
-     "shell.execute_reply": "2026-03-19T21:28:40.481889Z"
+     "iopub.execute_input": "2026-03-20T00:27:25.494719Z",
+     "iopub.status.busy": "2026-03-20T00:27:25.494242Z",
+     "iopub.status.idle": "2026-03-20T00:27:26.593638Z",
+     "shell.execute_reply": "2026-03-20T00:27:26.593341Z"
     }
    },
    "outputs": [],
@@ -84,10 +84,10 @@
    "id": "ad2b260b",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:28:40.483748Z",
-     "iopub.status.busy": "2026-03-19T21:28:40.483605Z",
-     "iopub.status.idle": "2026-03-19T21:28:41.705940Z",
-     "shell.execute_reply": "2026-03-19T21:28:41.705419Z"
+     "iopub.execute_input": "2026-03-20T00:27:26.595547Z",
+     "iopub.status.busy": "2026-03-20T00:27:26.595246Z",
+     "iopub.status.idle": "2026-03-20T00:27:27.145541Z",
+     "shell.execute_reply": "2026-03-20T00:27:27.144490Z"
     }
    },
    "outputs": [],
@@ -118,16 +118,16 @@
    "id": "e68f86ec",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:28:41.708217Z",
-     "iopub.status.busy": "2026-03-19T21:28:41.708045Z",
-     "iopub.status.idle": "2026-03-19T21:28:41.717057Z",
-     "shell.execute_reply": "2026-03-19T21:28:41.716703Z"
+     "iopub.execute_input": "2026-03-20T00:27:27.150888Z",
+     "iopub.status.busy": "2026-03-20T00:27:27.150527Z",
+     "iopub.status.idle": "2026-03-20T00:27:27.162843Z",
+     "shell.execute_reply": "2026-03-20T00:27:27.162398Z"
     }
    },
    "outputs": [],
    "source": [
     "# Read the image with Kornia\n",
-    "data = kornia.image_to_tensor(np.array(Image.open(\"squirrel.jpg\").convert(\"RGB\")))[None, ...]  # BxCxHxW"
+    "data = kornia.image_to_tensor(np.array(Image.open(\"squirrel.jpg\").convert(\"RGB\"))).float()[None, ...] / 255.0  # BxCxHxW"
    ]
   },
   {
@@ -144,10 +144,10 @@
    "id": "c8c80a7c",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:28:41.718934Z",
-     "iopub.status.busy": "2026-03-19T21:28:41.718785Z",
-     "iopub.status.idle": "2026-03-19T21:28:41.725420Z",
-     "shell.execute_reply": "2026-03-19T21:28:41.724887Z"
+     "iopub.execute_input": "2026-03-20T00:27:27.164969Z",
+     "iopub.status.busy": "2026-03-20T00:27:27.164803Z",
+     "iopub.status.idle": "2026-03-20T00:27:27.171552Z",
+     "shell.execute_reply": "2026-03-20T00:27:27.171207Z"
     }
    },
    "outputs": [],
@@ -163,10 +163,10 @@
    "id": "94a72aa3",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:28:41.727489Z",
-     "iopub.status.busy": "2026-03-19T21:28:41.727330Z",
-     "iopub.status.idle": "2026-03-19T21:28:41.729542Z",
-     "shell.execute_reply": "2026-03-19T21:28:41.729239Z"
+     "iopub.execute_input": "2026-03-20T00:27:27.173618Z",
+     "iopub.status.busy": "2026-03-20T00:27:27.173429Z",
+     "iopub.status.idle": "2026-03-20T00:27:27.176067Z",
+     "shell.execute_reply": "2026-03-20T00:27:27.175749Z"
     }
    },
    "outputs": [],
@@ -195,10 +195,10 @@
    "id": "98fc06a7",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:28:41.731075Z",
-     "iopub.status.busy": "2026-03-19T21:28:41.730940Z",
-     "iopub.status.idle": "2026-03-19T21:28:41.823336Z",
-     "shell.execute_reply": "2026-03-19T21:28:41.823047Z"
+     "iopub.execute_input": "2026-03-20T00:27:27.177649Z",
+     "iopub.status.busy": "2026-03-20T00:27:27.177497Z",
+     "iopub.status.idle": "2026-03-20T00:27:27.284884Z",
+     "shell.execute_reply": "2026-03-20T00:27:27.284579Z"
     }
    },
    "outputs": [],
diff --git a/nbs/visual_prompter.ipynb b/nbs/visual_prompter.ipynb
index 8e5acb5..cd4cb79 100644
--- a/nbs/visual_prompter.ipynb
+++ b/nbs/visual_prompter.ipynb
@@ -118,21 +118,7 @@
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": [
-    "from __future__ import annotations\n",
-    "\n",
-    "import os\n",
-    "\n",
-    "import matplotlib.pyplot as plt\n",
-    "import numpy as np\n",
-    "import torch\n",
-    "from kornia.contrib.models.sam import SamConfig\n",
-    "from kornia.contrib.visual_prompter import VisualPrompter\n",
-    "from kornia.geometry.boxes import Boxes\n",
-    "from kornia.geometry.keypoints import Keypoints\n",
-    "from kornia.utils import get_cuda_or_mps_device_if_available, tensor_to_image\n",
-    "from PIL import Image"
-   ]
+   "source": "from __future__ import annotations\n\nimport os\n\nimport kornia as K\nimport matplotlib.pyplot as plt\nimport numpy as np\nimport torch\nfrom kornia.contrib.visual_prompter import VisualPrompter\nfrom kornia.geometry.boxes import Boxes\nfrom kornia.geometry.keypoints import Keypoints\nfrom kornia.models.sam import SamConfig\nfrom kornia.utils import tensor_to_image\nfrom PIL import Image"
   },
   {
    "cell_type": "code",
@@ -140,7 +126,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "device = get_cuda_or_mps_device_if_available()\n",
+    "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"mps\" if torch.backends.mps.is_available() else \"cpu\")\n",
     "print(device)"
    ]
   },
@@ -278,7 +264,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from kornia.contrib.models import SegmentationResults\n",
+    "from kornia.models.structures import SegmentationResults\n",
     "\n",
     "\n",
     "def show_image(image: torch.Tensor):\n",
@@ -383,7 +369,7 @@
    "outputs": [],
    "source": [
     "# Load the image\n",
-    "image = K.image_to_tensor(np.array(Image.open(soccer_image_path).convert(\"RGB\"))).to(device)  # 3 x H x W\n",
+    "image = K.image_to_tensor(np.array(Image.open(soccer_image_path).convert(\"RGB\"))).float().to(device) / 255.0  # 3 x H x W\n",
     "\n",
     "# Display the loaded image\n",
     "show_image(image)"
@@ -666,7 +652,7 @@
    "outputs": [],
    "source": [
     "# Load the image\n",
-    "image = K.image_to_tensor(np.array(Image.open(soccer_image_path).convert(\"RGB\"))).to(device)  # 3 x H x W\n",
+    "image = K.image_to_tensor(np.array(Image.open(soccer_image_path).convert(\"RGB\"))).float().to(device) / 255.0  # 3 x H x W\n",
     "\n",
     "# Set the image\n",
     "prompter.set_image(image)"
@@ -761,7 +747,7 @@
    "outputs": [],
    "source": [
     "# Image\n",
-    "image = K.image_to_tensor(np.array(Image.open(car_image_path).convert(\"RGB\"))).to(device)\n",
+    "image = K.image_to_tensor(np.array(Image.open(car_image_path).convert(\"RGB\"))).float().to(device) / 255.0\n",
     "\n",
     "# Set the image\n",
     "prompter.set_image(image)"
@@ -948,7 +934,7 @@
    "outputs": [],
    "source": [
     "# Image\n",
-    "image = K.image_to_tensor(np.array(Image.open(satellite_image_path).convert(\"RGB\"))).to(device)\n",
+    "image = K.image_to_tensor(np.array(Image.open(satellite_image_path).convert(\"RGB\"))).float().to(device) / 255.0\n",
     "\n",
     "# Set the image\n",
     "prompter.set_image(image)"
@@ -1067,4 +1053,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}
\ No newline at end of file
diff --git a/nbs/warp_perspective.ipynb b/nbs/warp_perspective.ipynb
index af8ebdf..fdfe960 100644
--- a/nbs/warp_perspective.ipynb
+++ b/nbs/warp_perspective.ipynb
@@ -48,10 +48,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:30:48.938237Z",
-     "iopub.status.busy": "2026-03-19T21:30:48.937608Z",
-     "iopub.status.idle": "2026-03-19T21:30:49.768529Z",
-     "shell.execute_reply": "2026-03-19T21:30:49.767442Z"
+     "iopub.execute_input": "2026-03-20T00:27:32.779648Z",
+     "iopub.status.busy": "2026-03-20T00:27:32.779148Z",
+     "iopub.status.idle": "2026-03-20T00:27:33.446563Z",
+     "shell.execute_reply": "2026-03-20T00:27:33.445769Z"
     }
    },
    "outputs": [],
@@ -88,10 +88,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:30:49.800742Z",
-     "iopub.status.busy": "2026-03-19T21:30:49.800554Z",
-     "iopub.status.idle": "2026-03-19T21:30:50.856155Z",
-     "shell.execute_reply": "2026-03-19T21:30:50.855817Z"
+     "iopub.execute_input": "2026-03-20T00:27:33.477439Z",
+     "iopub.status.busy": "2026-03-20T00:27:33.477236Z",
+     "iopub.status.idle": "2026-03-20T00:27:34.532213Z",
+     "shell.execute_reply": "2026-03-20T00:27:34.531817Z"
     }
    },
    "outputs": [],
@@ -109,15 +109,15 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:30:50.857683Z",
-     "iopub.status.busy": "2026-03-19T21:30:50.857560Z",
-     "iopub.status.idle": "2026-03-19T21:30:50.867069Z",
-     "shell.execute_reply": "2026-03-19T21:30:50.866801Z"
+     "iopub.execute_input": "2026-03-20T00:27:34.534270Z",
+     "iopub.status.busy": "2026-03-20T00:27:34.534055Z",
+     "iopub.status.idle": "2026-03-20T00:27:34.545956Z",
+     "shell.execute_reply": "2026-03-20T00:27:34.545607Z"
     }
    },
    "outputs": [],
    "source": [
-    "img = K.image_to_tensor(np.array(Image.open(\"bruce.png\").convert(\"RGB\")))[None, ...]  # BxCxHxW\n",
+    "img = K.image_to_tensor(np.array(Image.open(\"bruce.png\").convert(\"RGB\"))).float()[None, ...] / 255.0  # BxCxHxW\n",
     "print(img.shape)"
    ]
   },
@@ -133,10 +133,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:30:50.868451Z",
-     "iopub.status.busy": "2026-03-19T21:30:50.868362Z",
-     "iopub.status.idle": "2026-03-19T21:30:50.875405Z",
-     "shell.execute_reply": "2026-03-19T21:30:50.875128Z"
+     "iopub.execute_input": "2026-03-20T00:27:34.547416Z",
+     "iopub.status.busy": "2026-03-20T00:27:34.547308Z",
+     "iopub.status.idle": "2026-03-20T00:27:34.552260Z",
+     "shell.execute_reply": "2026-03-20T00:27:34.552001Z"
     }
    },
    "outputs": [],
@@ -186,10 +186,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:30:50.876845Z",
-     "iopub.status.busy": "2026-03-19T21:30:50.876750Z",
-     "iopub.status.idle": "2026-03-19T21:30:50.954115Z",
-     "shell.execute_reply": "2026-03-19T21:30:50.953795Z"
+     "iopub.execute_input": "2026-03-20T00:27:34.553640Z",
+     "iopub.status.busy": "2026-03-20T00:27:34.553548Z",
+     "iopub.status.idle": "2026-03-20T00:27:34.641700Z",
+     "shell.execute_reply": "2026-03-20T00:27:34.641305Z"
     }
    },
    "outputs": [],
diff --git a/nbs/zca_whitening.ipynb b/nbs/zca_whitening.ipynb
index fba0944..a18d8b6 100644
--- a/nbs/zca_whitening.ipynb
+++ b/nbs/zca_whitening.ipynb
@@ -46,10 +46,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:30:52.996700Z",
-     "iopub.status.busy": "2026-03-19T21:30:52.996057Z",
-     "iopub.status.idle": "2026-03-19T21:30:54.459891Z",
-     "shell.execute_reply": "2026-03-19T21:30:54.459587Z"
+     "iopub.execute_input": "2026-03-20T00:27:36.742377Z",
+     "iopub.status.busy": "2026-03-20T00:27:36.741676Z",
+     "iopub.status.idle": "2026-03-20T00:27:38.336594Z",
+     "shell.execute_reply": "2026-03-20T00:27:38.336298Z"
     }
    },
    "outputs": [],
@@ -68,10 +68,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:30:54.461540Z",
-     "iopub.status.busy": "2026-03-19T21:30:54.461407Z",
-     "iopub.status.idle": "2026-03-19T21:30:54.463488Z",
-     "shell.execute_reply": "2026-03-19T21:30:54.463239Z"
+     "iopub.execute_input": "2026-03-20T00:27:38.338507Z",
+     "iopub.status.busy": "2026-03-20T00:27:38.338335Z",
+     "iopub.status.idle": "2026-03-20T00:27:38.340778Z",
+     "shell.execute_reply": "2026-03-20T00:27:38.340415Z"
     }
    },
    "outputs": [],
@@ -95,10 +95,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:30:54.480425Z",
-     "iopub.status.busy": "2026-03-19T21:30:54.480302Z",
-     "iopub.status.idle": "2026-03-19T21:31:10.433113Z",
-     "shell.execute_reply": "2026-03-19T21:31:10.432041Z"
+     "iopub.execute_input": "2026-03-20T00:27:38.358740Z",
+     "iopub.status.busy": "2026-03-20T00:27:38.358575Z",
+     "iopub.status.idle": "2026-03-20T00:27:38.377426Z",
+     "shell.execute_reply": "2026-03-20T00:27:38.377132Z"
     }
    },
    "outputs": [],
@@ -124,10 +124,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:31:10.435862Z",
-     "iopub.status.busy": "2026-03-19T21:31:10.435660Z",
-     "iopub.status.idle": "2026-03-19T21:31:11.970576Z",
-     "shell.execute_reply": "2026-03-19T21:31:11.970202Z"
+     "iopub.execute_input": "2026-03-20T00:27:38.379042Z",
+     "iopub.status.busy": "2026-03-20T00:27:38.378924Z",
+     "iopub.status.idle": "2026-03-20T00:27:40.106044Z",
+     "shell.execute_reply": "2026-03-20T00:27:40.105406Z"
     }
    },
    "outputs": [],
@@ -151,10 +151,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:31:11.972289Z",
-     "iopub.status.busy": "2026-03-19T21:31:11.972182Z",
-     "iopub.status.idle": "2026-03-19T21:31:12.121713Z",
-     "shell.execute_reply": "2026-03-19T21:31:12.121379Z"
+     "iopub.execute_input": "2026-03-20T00:27:40.108286Z",
+     "iopub.status.busy": "2026-03-20T00:27:40.108117Z",
+     "iopub.status.idle": "2026-03-20T00:27:40.255889Z",
+     "shell.execute_reply": "2026-03-20T00:27:40.255528Z"
     }
    },
    "outputs": [],
@@ -175,10 +175,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:31:12.123299Z",
-     "iopub.status.busy": "2026-03-19T21:31:12.123200Z",
-     "iopub.status.idle": "2026-03-19T21:31:12.236657Z",
-     "shell.execute_reply": "2026-03-19T21:31:12.236356Z"
+     "iopub.execute_input": "2026-03-20T00:27:40.257575Z",
+     "iopub.status.busy": "2026-03-20T00:27:40.257464Z",
+     "iopub.status.idle": "2026-03-20T00:27:40.370337Z",
+     "shell.execute_reply": "2026-03-20T00:27:40.369985Z"
     }
    },
    "outputs": [],
@@ -216,10 +216,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:31:12.238769Z",
-     "iopub.status.busy": "2026-03-19T21:31:12.238635Z",
-     "iopub.status.idle": "2026-03-19T21:32:06.797605Z",
-     "shell.execute_reply": "2026-03-19T21:32:06.797246Z"
+     "iopub.execute_input": "2026-03-20T00:27:40.372179Z",
+     "iopub.status.busy": "2026-03-20T00:27:40.371926Z",
+     "iopub.status.idle": "2026-03-20T00:27:43.285850Z",
+     "shell.execute_reply": "2026-03-20T00:27:43.285544Z"
     }
    },
    "outputs": [],
@@ -250,10 +250,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:32:06.799756Z",
-     "iopub.status.busy": "2026-03-19T21:32:06.799628Z",
-     "iopub.status.idle": "2026-03-19T21:32:11.823397Z",
-     "shell.execute_reply": "2026-03-19T21:32:11.823074Z"
+     "iopub.execute_input": "2026-03-20T00:27:43.287479Z",
+     "iopub.status.busy": "2026-03-20T00:27:43.287389Z",
+     "iopub.status.idle": "2026-03-20T00:27:48.692898Z",
+     "shell.execute_reply": "2026-03-20T00:27:48.692373Z"
     }
    },
    "outputs": [],
@@ -276,10 +276,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:32:11.825130Z",
-     "iopub.status.busy": "2026-03-19T21:32:11.825039Z",
-     "iopub.status.idle": "2026-03-19T21:32:12.011025Z",
-     "shell.execute_reply": "2026-03-19T21:32:12.010295Z"
+     "iopub.execute_input": "2026-03-20T00:27:48.694903Z",
+     "iopub.status.busy": "2026-03-20T00:27:48.694633Z",
+     "iopub.status.idle": "2026-03-20T00:27:48.900207Z",
+     "shell.execute_reply": "2026-03-20T00:27:48.899577Z"
     }
    },
    "outputs": [],
@@ -335,10 +335,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:32:12.018494Z",
-     "iopub.status.busy": "2026-03-19T21:32:12.018364Z",
-     "iopub.status.idle": "2026-03-19T21:32:12.062331Z",
-     "shell.execute_reply": "2026-03-19T21:32:12.062041Z"
+     "iopub.execute_input": "2026-03-20T00:27:48.908773Z",
+     "iopub.status.busy": "2026-03-20T00:27:48.908540Z",
+     "iopub.status.idle": "2026-03-20T00:27:48.957966Z",
+     "shell.execute_reply": "2026-03-20T00:27:48.957621Z"
     }
    },
    "outputs": [],
@@ -374,10 +374,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:32:12.063924Z",
-     "iopub.status.busy": "2026-03-19T21:32:12.063807Z",
-     "iopub.status.idle": "2026-03-19T21:32:12.065686Z",
-     "shell.execute_reply": "2026-03-19T21:32:12.065420Z"
+     "iopub.execute_input": "2026-03-20T00:27:48.959830Z",
+     "iopub.status.busy": "2026-03-20T00:27:48.959672Z",
+     "iopub.status.idle": "2026-03-20T00:27:48.962086Z",
+     "shell.execute_reply": "2026-03-20T00:27:48.961633Z"
     }
    },
    "outputs": [],
@@ -398,10 +398,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:32:12.067154Z",
-     "iopub.status.busy": "2026-03-19T21:32:12.067036Z",
-     "iopub.status.idle": "2026-03-19T21:32:12.108889Z",
-     "shell.execute_reply": "2026-03-19T21:32:12.108579Z"
+     "iopub.execute_input": "2026-03-20T00:27:48.963958Z",
+     "iopub.status.busy": "2026-03-20T00:27:48.963797Z",
+     "iopub.status.idle": "2026-03-20T00:27:49.020525Z",
+     "shell.execute_reply": "2026-03-20T00:27:49.020166Z"
     }
    },
    "outputs": [],
@@ -435,10 +435,10 @@
    "execution_count": null,
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2026-03-19T21:32:12.110436Z",
-     "iopub.status.busy": "2026-03-19T21:32:12.110339Z",
-     "iopub.status.idle": "2026-03-19T21:32:12.149455Z",
-     "shell.execute_reply": "2026-03-19T21:32:12.149158Z"
+     "iopub.execute_input": "2026-03-20T00:27:49.022496Z",
+     "iopub.status.busy": "2026-03-20T00:27:49.022289Z",
+     "iopub.status.idle": "2026-03-20T00:27:49.062781Z",
+     "shell.execute_reply": "2026-03-20T00:27:49.062291Z"
     }
    },
    "outputs": [],