diff --git a/docs/api/backends/backend.md b/docs/api/backends/backend.md
index 924f947c..809542db 100644
--- a/docs/api/backends/backend.md
+++ b/docs/api/backends/backend.md
@@ -14,4 +14,4 @@ for backend in hk.BackendFactory.list():
     print(f"Backend name: {backend} - {hk.BackendFactory.get(backend)}")
 ```
 
-::: neuralspot_edge.utils.ItemFactory
+<!-- ::: neuralspot_edge.utils.ItemFactory -->
diff --git a/docs/api/datasets/factory.md b/docs/api/datasets/factory.md
index 2c4ff36a..09980905 100644
--- a/docs/api/datasets/factory.md
+++ b/docs/api/datasets/factory.md
@@ -13,4 +13,4 @@ for dataset in hk.DatasetFactory.list():
     print(f"Dataset name: {dataset} - {hk.DatasetFactory.get(dataset)}")
 ```
 
-::: neuralspot_edge.utils.ItemFactory
+<!-- ::: neuralspot_edge.utils.ItemFactory -->
diff --git a/docs/api/tasks/factory.md b/docs/api/tasks/factory.md
index 4d3ed6fd..f260fd40 100644
--- a/docs/api/tasks/factory.md
+++ b/docs/api/tasks/factory.md
@@ -11,4 +11,4 @@ for model in hk.TaskFactory.list():
     print(f"Task name: {model} - {hk.TaskFactory.get(model)}")
 ```
 
-::: neuralspot_edge.utils.ItemFactory
+<!-- ::: neuralspot_edge.utils.ItemFactory -->
diff --git a/docs/guides/byot.ipynb b/docs/guides/byot.ipynb
index 1ce0f5af..d8d31fac 100644
--- a/docs/guides/byot.ipynb
+++ b/docs/guides/byot.ipynb
@@ -8,12 +8,12 @@
     "<div class=\"grid cards\" markdown>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 488 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M488 261.8C488 403.3 391.1 504 248 504 110.8 504 0 393.2 0 256S110.8 8 248 8c66.8 0 123 24.5 166.3 64.9l-67.5 64.9C258.5 52.6 94.3 116.6 94.3 256c0 86.5 69.1 156.6 153.7 156.6 98.2 0 135-70.4 140.8-106.9H248v-85.3h236.1c2.3 12.7 3.9 24.9 3.9 41.4z\"/></svg>\n",
-    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/canary/docs/guides/byot.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/main/docs/guides/byot.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    View in Colab\n",
     "</a>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 496 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M165.9 397.4c0 2-2.3 3.6-5.2 3.6-3.3 .3-5.6-1.3-5.6-3.6 0-2 2.3-3.6 5.2-3.6 3-.3 5.6 1.3 5.6 3.6zm-31.1-4.5c-.7 2 1.3 4.3 4.3 4.9 2.6 1 5.6 0 6.2-2s-1.3-4.3-4.3-5.2c-2.6-.7-5.5 .3-6.2 2.3zm44.2-1.7c-2.9 .7-4.9 2.6-4.6 4.9 .3 2 2.9 3.3 5.9 2.6 2.9-.7 4.9-2.6 4.6-4.6-.3-1.9-3-3.2-5.9-2.9zM244.8 8C106.1 8 0 113.3 0 252c0 110.9 69.8 205.8 169.5 239.2 12.8 2.3 17.3-5.6 17.3-12.1 0-6.2-.3-40.4-.3-61.4 0 0-70 15-84.7-29.8 0 0-11.4-29.1-27.8-36.6 0 0-22.9-15.7 1.6-15.4 0 0 24.9 2 38.6 25.8 21.9 38.6 58.6 27.5 72.9 20.9 2.3-16 8.8-27.1 16-33.7-55.9-6.2-112.3-14.3-112.3-110.5 0-27.5 7.6-41.3 23.6-58.9-2.6-6.5-11.1-33.3 2.6-67.9 20.9-6.5 69 27 69 27 20-5.6 41.5-8.5 62.8-8.5s42.8 2.9 62.8 8.5c0 0 48.1-33.6 69-27 13.7 34.7 5.2 61.4 2.6 67.9 16 17.7 25.8 31.5 25.8 58.9 0 96.5-58.9 104.2-114.8 110.5 9.2 7.9 17 22.9 17 46.4 0 33.7-.3 75.4-.3 83.6 0 6.5 4.6 14.4 17.3 12.1C428.2 457.8 496 362.9 496 252 496 113.3 383.5 8 244.8 8zM97.2 352.9c-1.3 1-1 3.3 .7 5.2 1.6 1.6 3.9 2.3 5.2 1 1.3-1 1-3.3-.7-5.2-1.6-1.6-3.9-2.3-5.2-1zm-10.8-8.1c-.7 1.3 .3 2.9 2.3 3.9 1.6 1 3.6 .7 4.3-.7 .7-1.3-.3-2.9-2.3-3.9-2-.6-3.6-.3-4.3 .7zm32.4 35.6c-1.6 1.3-1 4.3 1.3 6.2 2.3 2.3 5.2 2.6 6.5 1 1.3-1.3 .7-4.3-1.3-6.2-2.2-2.3-5.2-2.6-6.5-1zm-11.4-14.7c-1.6 1-1.6 3.6 0 5.9 1.6 2.3 4.3 3.3 5.6 2.3 1.6-1.3 1.6-3.9 0-6.2-1.4-2.3-4-3.3-5.6-2z\"/></svg>\n",
-    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/canary/docs/guides/byot.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/main/docs/guides/byot.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    GitHub source\n",
     "</a>\n",
     "\n",
diff --git a/docs/guides/ecg-foundation-model.ipynb b/docs/guides/ecg-foundation-model.ipynb
index 61f8193a..3e973457 100644
--- a/docs/guides/ecg-foundation-model.ipynb
+++ b/docs/guides/ecg-foundation-model.ipynb
@@ -8,12 +8,12 @@
     "<div class=\"grid cards\" markdown>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 488 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M488 261.8C488 403.3 391.1 504 248 504 110.8 504 0 393.2 0 256S110.8 8 248 8c66.8 0 123 24.5 166.3 64.9l-67.5 64.9C258.5 52.6 94.3 116.6 94.3 256c0 86.5 69.1 156.6 153.7 156.6 98.2 0 135-70.4 140.8-106.9H248v-85.3h236.1c2.3 12.7 3.9 24.9 3.9 41.4z\"/></svg>\n",
-    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/canary/docs/guides/ecg-foundation-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/main/docs/guides/ecg-foundation-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    View in Colab\n",
     "</a>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 496 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M165.9 397.4c0 2-2.3 3.6-5.2 3.6-3.3 .3-5.6-1.3-5.6-3.6 0-2 2.3-3.6 5.2-3.6 3-.3 5.6 1.3 5.6 3.6zm-31.1-4.5c-.7 2 1.3 4.3 4.3 4.9 2.6 1 5.6 0 6.2-2s-1.3-4.3-4.3-5.2c-2.6-.7-5.5 .3-6.2 2.3zm44.2-1.7c-2.9 .7-4.9 2.6-4.6 4.9 .3 2 2.9 3.3 5.9 2.6 2.9-.7 4.9-2.6 4.6-4.6-.3-1.9-3-3.2-5.9-2.9zM244.8 8C106.1 8 0 113.3 0 252c0 110.9 69.8 205.8 169.5 239.2 12.8 2.3 17.3-5.6 17.3-12.1 0-6.2-.3-40.4-.3-61.4 0 0-70 15-84.7-29.8 0 0-11.4-29.1-27.8-36.6 0 0-22.9-15.7 1.6-15.4 0 0 24.9 2 38.6 25.8 21.9 38.6 58.6 27.5 72.9 20.9 2.3-16 8.8-27.1 16-33.7-55.9-6.2-112.3-14.3-112.3-110.5 0-27.5 7.6-41.3 23.6-58.9-2.6-6.5-11.1-33.3 2.6-67.9 20.9-6.5 69 27 69 27 20-5.6 41.5-8.5 62.8-8.5s42.8 2.9 62.8 8.5c0 0 48.1-33.6 69-27 13.7 34.7 5.2 61.4 2.6 67.9 16 17.7 25.8 31.5 25.8 58.9 0 96.5-58.9 104.2-114.8 110.5 9.2 7.9 17 22.9 17 46.4 0 33.7-.3 75.4-.3 83.6 0 6.5 4.6 14.4 17.3 12.1C428.2 457.8 496 362.9 496 252 496 113.3 383.5 8 244.8 8zM97.2 352.9c-1.3 1-1 3.3 .7 5.2 1.6 1.6 3.9 2.3 5.2 1 1.3-1 1-3.3-.7-5.2-1.6-1.6-3.9-2.3-5.2-1zm-10.8-8.1c-.7 1.3 .3 2.9 2.3 3.9 1.6 1 3.6 .7 4.3-.7 .7-1.3-.3-2.9-2.3-3.9-2-.6-3.6-.3-4.3 .7zm32.4 35.6c-1.6 1.3-1 4.3 1.3 6.2 2.3 2.3 5.2 2.6 6.5 1 1.3-1.3 .7-4.3-1.3-6.2-2.2-2.3-5.2-2.6-6.5-1zm-11.4-14.7c-1.6 1-1.6 3.6 0 5.9 1.6 2.3 4.3 3.3 5.6 2.3 1.6-1.3 1.6-3.9 0-6.2-1.4-2.3-4-3.3-5.6-2z\"/></svg>\n",
-    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/canary/docs/guides/ecg-foundation-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/main/docs/guides/ecg-foundation-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    GitHub source\n",
     "</a>\n",
     "\n",
diff --git a/docs/guides/train-arrhythmia-model.ipynb b/docs/guides/train-arrhythmia-model.ipynb
index c9eb5307..fbaef426 100644
--- a/docs/guides/train-arrhythmia-model.ipynb
+++ b/docs/guides/train-arrhythmia-model.ipynb
@@ -8,12 +8,12 @@
     "<div class=\"grid cards\" markdown>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 488 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M488 261.8C488 403.3 391.1 504 248 504 110.8 504 0 393.2 0 256S110.8 8 248 8c66.8 0 123 24.5 166.3 64.9l-67.5 64.9C258.5 52.6 94.3 116.6 94.3 256c0 86.5 69.1 156.6 153.7 156.6 98.2 0 135-70.4 140.8-106.9H248v-85.3h236.1c2.3 12.7 3.9 24.9 3.9 41.4z\"/></svg>\n",
-    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/canary/docs/guides/train-arrhythmia-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/main/docs/guides/train-arrhythmia-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    View in Colab\n",
     "</a>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 496 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M165.9 397.4c0 2-2.3 3.6-5.2 3.6-3.3 .3-5.6-1.3-5.6-3.6 0-2 2.3-3.6 5.2-3.6 3-.3 5.6 1.3 5.6 3.6zm-31.1-4.5c-.7 2 1.3 4.3 4.3 4.9 2.6 1 5.6 0 6.2-2s-1.3-4.3-4.3-5.2c-2.6-.7-5.5 .3-6.2 2.3zm44.2-1.7c-2.9 .7-4.9 2.6-4.6 4.9 .3 2 2.9 3.3 5.9 2.6 2.9-.7 4.9-2.6 4.6-4.6-.3-1.9-3-3.2-5.9-2.9zM244.8 8C106.1 8 0 113.3 0 252c0 110.9 69.8 205.8 169.5 239.2 12.8 2.3 17.3-5.6 17.3-12.1 0-6.2-.3-40.4-.3-61.4 0 0-70 15-84.7-29.8 0 0-11.4-29.1-27.8-36.6 0 0-22.9-15.7 1.6-15.4 0 0 24.9 2 38.6 25.8 21.9 38.6 58.6 27.5 72.9 20.9 2.3-16 8.8-27.1 16-33.7-55.9-6.2-112.3-14.3-112.3-110.5 0-27.5 7.6-41.3 23.6-58.9-2.6-6.5-11.1-33.3 2.6-67.9 20.9-6.5 69 27 69 27 20-5.6 41.5-8.5 62.8-8.5s42.8 2.9 62.8 8.5c0 0 48.1-33.6 69-27 13.7 34.7 5.2 61.4 2.6 67.9 16 17.7 25.8 31.5 25.8 58.9 0 96.5-58.9 104.2-114.8 110.5 9.2 7.9 17 22.9 17 46.4 0 33.7-.3 75.4-.3 83.6 0 6.5 4.6 14.4 17.3 12.1C428.2 457.8 496 362.9 496 252 496 113.3 383.5 8 244.8 8zM97.2 352.9c-1.3 1-1 3.3 .7 5.2 1.6 1.6 3.9 2.3 5.2 1 1.3-1 1-3.3-.7-5.2-1.6-1.6-3.9-2.3-5.2-1zm-10.8-8.1c-.7 1.3 .3 2.9 2.3 3.9 1.6 1 3.6 .7 4.3-.7 .7-1.3-.3-2.9-2.3-3.9-2-.6-3.6-.3-4.3 .7zm32.4 35.6c-1.6 1.3-1 4.3 1.3 6.2 2.3 2.3 5.2 2.6 6.5 1 1.3-1.3 .7-4.3-1.3-6.2-2.2-2.3-5.2-2.6-6.5-1zm-11.4-14.7c-1.6 1-1.6 3.6 0 5.9 1.6 2.3 4.3 3.3 5.6 2.3 1.6-1.3 1.6-3.9 0-6.2-1.4-2.3-4-3.3-5.6-2z\"/></svg>\n",
-    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/canary/docs/guides/train-arrhythmia-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/main/docs/guides/train-arrhythmia-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    GitHub source\n",
     "</a>\n",
     "\n",
diff --git a/docs/guides/train-ecg-denoiser.ipynb b/docs/guides/train-ecg-denoiser.ipynb
index 71de1d5e..b450b12d 100644
--- a/docs/guides/train-ecg-denoiser.ipynb
+++ b/docs/guides/train-ecg-denoiser.ipynb
@@ -8,12 +8,12 @@
     "<div class=\"grid cards\" markdown>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 488 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M488 261.8C488 403.3 391.1 504 248 504 110.8 504 0 393.2 0 256S110.8 8 248 8c66.8 0 123 24.5 166.3 64.9l-67.5 64.9C258.5 52.6 94.3 116.6 94.3 256c0 86.5 69.1 156.6 153.7 156.6 98.2 0 135-70.4 140.8-106.9H248v-85.3h236.1c2.3 12.7 3.9 24.9 3.9 41.4z\"/></svg>\n",
-    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/canary/docs/guides/train-ecg-denoiser.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/main/docs/guides/train-ecg-denoiser.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    View in Colab\n",
     "</a>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 496 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M165.9 397.4c0 2-2.3 3.6-5.2 3.6-3.3 .3-5.6-1.3-5.6-3.6 0-2 2.3-3.6 5.2-3.6 3-.3 5.6 1.3 5.6 3.6zm-31.1-4.5c-.7 2 1.3 4.3 4.3 4.9 2.6 1 5.6 0 6.2-2s-1.3-4.3-4.3-5.2c-2.6-.7-5.5 .3-6.2 2.3zm44.2-1.7c-2.9 .7-4.9 2.6-4.6 4.9 .3 2 2.9 3.3 5.9 2.6 2.9-.7 4.9-2.6 4.6-4.6-.3-1.9-3-3.2-5.9-2.9zM244.8 8C106.1 8 0 113.3 0 252c0 110.9 69.8 205.8 169.5 239.2 12.8 2.3 17.3-5.6 17.3-12.1 0-6.2-.3-40.4-.3-61.4 0 0-70 15-84.7-29.8 0 0-11.4-29.1-27.8-36.6 0 0-22.9-15.7 1.6-15.4 0 0 24.9 2 38.6 25.8 21.9 38.6 58.6 27.5 72.9 20.9 2.3-16 8.8-27.1 16-33.7-55.9-6.2-112.3-14.3-112.3-110.5 0-27.5 7.6-41.3 23.6-58.9-2.6-6.5-11.1-33.3 2.6-67.9 20.9-6.5 69 27 69 27 20-5.6 41.5-8.5 62.8-8.5s42.8 2.9 62.8 8.5c0 0 48.1-33.6 69-27 13.7 34.7 5.2 61.4 2.6 67.9 16 17.7 25.8 31.5 25.8 58.9 0 96.5-58.9 104.2-114.8 110.5 9.2 7.9 17 22.9 17 46.4 0 33.7-.3 75.4-.3 83.6 0 6.5 4.6 14.4 17.3 12.1C428.2 457.8 496 362.9 496 252 496 113.3 383.5 8 244.8 8zM97.2 352.9c-1.3 1-1 3.3 .7 5.2 1.6 1.6 3.9 2.3 5.2 1 1.3-1 1-3.3-.7-5.2-1.6-1.6-3.9-2.3-5.2-1zm-10.8-8.1c-.7 1.3 .3 2.9 2.3 3.9 1.6 1 3.6 .7 4.3-.7 .7-1.3-.3-2.9-2.3-3.9-2-.6-3.6-.3-4.3 .7zm32.4 35.6c-1.6 1.3-1 4.3 1.3 6.2 2.3 2.3 5.2 2.6 6.5 1 1.3-1.3 .7-4.3-1.3-6.2-2.2-2.3-5.2-2.6-6.5-1zm-11.4-14.7c-1.6 1-1.6 3.6 0 5.9 1.6 2.3 4.3 3.3 5.6 2.3 1.6-1.3 1.6-3.9 0-6.2-1.4-2.3-4-3.3-5.6-2z\"/></svg>\n",
-    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/canary/docs/guides/train-ecg-denoiser.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/main/docs/guides/train-ecg-denoiser.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    GitHub source\n",
     "</a>\n",
     "\n",
diff --git a/docs/guides/train-ecg-segmentation.ipynb b/docs/guides/train-ecg-segmentation.ipynb
index 7ed6ba14..e688dba0 100644
--- a/docs/guides/train-ecg-segmentation.ipynb
+++ b/docs/guides/train-ecg-segmentation.ipynb
@@ -8,12 +8,12 @@
     "<div class=\"grid cards\" markdown>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 488 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M488 261.8C488 403.3 391.1 504 248 504 110.8 504 0 393.2 0 256S110.8 8 248 8c66.8 0 123 24.5 166.3 64.9l-67.5 64.9C258.5 52.6 94.3 116.6 94.3 256c0 86.5 69.1 156.6 153.7 156.6 98.2 0 135-70.4 140.8-106.9H248v-85.3h236.1c2.3 12.7 3.9 24.9 3.9 41.4z\"/></svg>\n",
-    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/canary/docs/guides/train-ecg-segmentation.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/main/docs/guides/train-ecg-segmentation.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    View in Colab\n",
     "</a>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 496 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M165.9 397.4c0 2-2.3 3.6-5.2 3.6-3.3 .3-5.6-1.3-5.6-3.6 0-2 2.3-3.6 5.2-3.6 3-.3 5.6 1.3 5.6 3.6zm-31.1-4.5c-.7 2 1.3 4.3 4.3 4.9 2.6 1 5.6 0 6.2-2s-1.3-4.3-4.3-5.2c-2.6-.7-5.5 .3-6.2 2.3zm44.2-1.7c-2.9 .7-4.9 2.6-4.6 4.9 .3 2 2.9 3.3 5.9 2.6 2.9-.7 4.9-2.6 4.6-4.6-.3-1.9-3-3.2-5.9-2.9zM244.8 8C106.1 8 0 113.3 0 252c0 110.9 69.8 205.8 169.5 239.2 12.8 2.3 17.3-5.6 17.3-12.1 0-6.2-.3-40.4-.3-61.4 0 0-70 15-84.7-29.8 0 0-11.4-29.1-27.8-36.6 0 0-22.9-15.7 1.6-15.4 0 0 24.9 2 38.6 25.8 21.9 38.6 58.6 27.5 72.9 20.9 2.3-16 8.8-27.1 16-33.7-55.9-6.2-112.3-14.3-112.3-110.5 0-27.5 7.6-41.3 23.6-58.9-2.6-6.5-11.1-33.3 2.6-67.9 20.9-6.5 69 27 69 27 20-5.6 41.5-8.5 62.8-8.5s42.8 2.9 62.8 8.5c0 0 48.1-33.6 69-27 13.7 34.7 5.2 61.4 2.6 67.9 16 17.7 25.8 31.5 25.8 58.9 0 96.5-58.9 104.2-114.8 110.5 9.2 7.9 17 22.9 17 46.4 0 33.7-.3 75.4-.3 83.6 0 6.5 4.6 14.4 17.3 12.1C428.2 457.8 496 362.9 496 252 496 113.3 383.5 8 244.8 8zM97.2 352.9c-1.3 1-1 3.3 .7 5.2 1.6 1.6 3.9 2.3 5.2 1 1.3-1 1-3.3-.7-5.2-1.6-1.6-3.9-2.3-5.2-1zm-10.8-8.1c-.7 1.3 .3 2.9 2.3 3.9 1.6 1 3.6 .7 4.3-.7 .7-1.3-.3-2.9-2.3-3.9-2-.6-3.6-.3-4.3 .7zm32.4 35.6c-1.6 1.3-1 4.3 1.3 6.2 2.3 2.3 5.2 2.6 6.5 1 1.3-1.3 .7-4.3-1.3-6.2-2.2-2.3-5.2-2.6-6.5-1zm-11.4-14.7c-1.6 1-1.6 3.6 0 5.9 1.6 2.3 4.3 3.3 5.6 2.3 1.6-1.3 1.6-3.9 0-6.2-1.4-2.3-4-3.3-5.6-2z\"/></svg>\n",
-    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/canary/docs/guides/train-ecg-segmentation.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/main/docs/guides/train-ecg-segmentation.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    GitHub source\n",
     "</a>\n",
     "\n",
diff --git a/notebooks/byot.ipynb b/notebooks/byot.ipynb
index 1ce0f5af..d8d31fac 100644
--- a/notebooks/byot.ipynb
+++ b/notebooks/byot.ipynb
@@ -8,12 +8,12 @@
     "<div class=\"grid cards\" markdown>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 488 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M488 261.8C488 403.3 391.1 504 248 504 110.8 504 0 393.2 0 256S110.8 8 248 8c66.8 0 123 24.5 166.3 64.9l-67.5 64.9C258.5 52.6 94.3 116.6 94.3 256c0 86.5 69.1 156.6 153.7 156.6 98.2 0 135-70.4 140.8-106.9H248v-85.3h236.1c2.3 12.7 3.9 24.9 3.9 41.4z\"/></svg>\n",
-    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/canary/docs/guides/byot.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/main/docs/guides/byot.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    View in Colab\n",
     "</a>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 496 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M165.9 397.4c0 2-2.3 3.6-5.2 3.6-3.3 .3-5.6-1.3-5.6-3.6 0-2 2.3-3.6 5.2-3.6 3-.3 5.6 1.3 5.6 3.6zm-31.1-4.5c-.7 2 1.3 4.3 4.3 4.9 2.6 1 5.6 0 6.2-2s-1.3-4.3-4.3-5.2c-2.6-.7-5.5 .3-6.2 2.3zm44.2-1.7c-2.9 .7-4.9 2.6-4.6 4.9 .3 2 2.9 3.3 5.9 2.6 2.9-.7 4.9-2.6 4.6-4.6-.3-1.9-3-3.2-5.9-2.9zM244.8 8C106.1 8 0 113.3 0 252c0 110.9 69.8 205.8 169.5 239.2 12.8 2.3 17.3-5.6 17.3-12.1 0-6.2-.3-40.4-.3-61.4 0 0-70 15-84.7-29.8 0 0-11.4-29.1-27.8-36.6 0 0-22.9-15.7 1.6-15.4 0 0 24.9 2 38.6 25.8 21.9 38.6 58.6 27.5 72.9 20.9 2.3-16 8.8-27.1 16-33.7-55.9-6.2-112.3-14.3-112.3-110.5 0-27.5 7.6-41.3 23.6-58.9-2.6-6.5-11.1-33.3 2.6-67.9 20.9-6.5 69 27 69 27 20-5.6 41.5-8.5 62.8-8.5s42.8 2.9 62.8 8.5c0 0 48.1-33.6 69-27 13.7 34.7 5.2 61.4 2.6 67.9 16 17.7 25.8 31.5 25.8 58.9 0 96.5-58.9 104.2-114.8 110.5 9.2 7.9 17 22.9 17 46.4 0 33.7-.3 75.4-.3 83.6 0 6.5 4.6 14.4 17.3 12.1C428.2 457.8 496 362.9 496 252 496 113.3 383.5 8 244.8 8zM97.2 352.9c-1.3 1-1 3.3 .7 5.2 1.6 1.6 3.9 2.3 5.2 1 1.3-1 1-3.3-.7-5.2-1.6-1.6-3.9-2.3-5.2-1zm-10.8-8.1c-.7 1.3 .3 2.9 2.3 3.9 1.6 1 3.6 .7 4.3-.7 .7-1.3-.3-2.9-2.3-3.9-2-.6-3.6-.3-4.3 .7zm32.4 35.6c-1.6 1.3-1 4.3 1.3 6.2 2.3 2.3 5.2 2.6 6.5 1 1.3-1.3 .7-4.3-1.3-6.2-2.2-2.3-5.2-2.6-6.5-1zm-11.4-14.7c-1.6 1-1.6 3.6 0 5.9 1.6 2.3 4.3 3.3 5.6 2.3 1.6-1.3 1.6-3.9 0-6.2-1.4-2.3-4-3.3-5.6-2z\"/></svg>\n",
-    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/canary/docs/guides/byot.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/main/docs/guides/byot.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    GitHub source\n",
     "</a>\n",
     "\n",
diff --git a/notebooks/ecg-foundation-model.ipynb b/notebooks/ecg-foundation-model.ipynb
index 61f8193a..3e973457 100644
--- a/notebooks/ecg-foundation-model.ipynb
+++ b/notebooks/ecg-foundation-model.ipynb
@@ -8,12 +8,12 @@
     "<div class=\"grid cards\" markdown>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 488 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M488 261.8C488 403.3 391.1 504 248 504 110.8 504 0 393.2 0 256S110.8 8 248 8c66.8 0 123 24.5 166.3 64.9l-67.5 64.9C258.5 52.6 94.3 116.6 94.3 256c0 86.5 69.1 156.6 153.7 156.6 98.2 0 135-70.4 140.8-106.9H248v-85.3h236.1c2.3 12.7 3.9 24.9 3.9 41.4z\"/></svg>\n",
-    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/canary/docs/guides/ecg-foundation-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/main/docs/guides/ecg-foundation-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    View in Colab\n",
     "</a>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 496 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M165.9 397.4c0 2-2.3 3.6-5.2 3.6-3.3 .3-5.6-1.3-5.6-3.6 0-2 2.3-3.6 5.2-3.6 3-.3 5.6 1.3 5.6 3.6zm-31.1-4.5c-.7 2 1.3 4.3 4.3 4.9 2.6 1 5.6 0 6.2-2s-1.3-4.3-4.3-5.2c-2.6-.7-5.5 .3-6.2 2.3zm44.2-1.7c-2.9 .7-4.9 2.6-4.6 4.9 .3 2 2.9 3.3 5.9 2.6 2.9-.7 4.9-2.6 4.6-4.6-.3-1.9-3-3.2-5.9-2.9zM244.8 8C106.1 8 0 113.3 0 252c0 110.9 69.8 205.8 169.5 239.2 12.8 2.3 17.3-5.6 17.3-12.1 0-6.2-.3-40.4-.3-61.4 0 0-70 15-84.7-29.8 0 0-11.4-29.1-27.8-36.6 0 0-22.9-15.7 1.6-15.4 0 0 24.9 2 38.6 25.8 21.9 38.6 58.6 27.5 72.9 20.9 2.3-16 8.8-27.1 16-33.7-55.9-6.2-112.3-14.3-112.3-110.5 0-27.5 7.6-41.3 23.6-58.9-2.6-6.5-11.1-33.3 2.6-67.9 20.9-6.5 69 27 69 27 20-5.6 41.5-8.5 62.8-8.5s42.8 2.9 62.8 8.5c0 0 48.1-33.6 69-27 13.7 34.7 5.2 61.4 2.6 67.9 16 17.7 25.8 31.5 25.8 58.9 0 96.5-58.9 104.2-114.8 110.5 9.2 7.9 17 22.9 17 46.4 0 33.7-.3 75.4-.3 83.6 0 6.5 4.6 14.4 17.3 12.1C428.2 457.8 496 362.9 496 252 496 113.3 383.5 8 244.8 8zM97.2 352.9c-1.3 1-1 3.3 .7 5.2 1.6 1.6 3.9 2.3 5.2 1 1.3-1 1-3.3-.7-5.2-1.6-1.6-3.9-2.3-5.2-1zm-10.8-8.1c-.7 1.3 .3 2.9 2.3 3.9 1.6 1 3.6 .7 4.3-.7 .7-1.3-.3-2.9-2.3-3.9-2-.6-3.6-.3-4.3 .7zm32.4 35.6c-1.6 1.3-1 4.3 1.3 6.2 2.3 2.3 5.2 2.6 6.5 1 1.3-1.3 .7-4.3-1.3-6.2-2.2-2.3-5.2-2.6-6.5-1zm-11.4-14.7c-1.6 1-1.6 3.6 0 5.9 1.6 2.3 4.3 3.3 5.6 2.3 1.6-1.3 1.6-3.9 0-6.2-1.4-2.3-4-3.3-5.6-2z\"/></svg>\n",
-    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/canary/docs/guides/ecg-foundation-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/main/docs/guides/ecg-foundation-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    GitHub source\n",
     "</a>\n",
     "\n",
diff --git a/notebooks/train-arrhythmia-model.ipynb b/notebooks/train-arrhythmia-model.ipynb
index c9eb5307..fbaef426 100644
--- a/notebooks/train-arrhythmia-model.ipynb
+++ b/notebooks/train-arrhythmia-model.ipynb
@@ -8,12 +8,12 @@
     "<div class=\"grid cards\" markdown>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 488 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M488 261.8C488 403.3 391.1 504 248 504 110.8 504 0 393.2 0 256S110.8 8 248 8c66.8 0 123 24.5 166.3 64.9l-67.5 64.9C258.5 52.6 94.3 116.6 94.3 256c0 86.5 69.1 156.6 153.7 156.6 98.2 0 135-70.4 140.8-106.9H248v-85.3h236.1c2.3 12.7 3.9 24.9 3.9 41.4z\"/></svg>\n",
-    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/canary/docs/guides/train-arrhythmia-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/main/docs/guides/train-arrhythmia-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    View in Colab\n",
     "</a>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 496 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M165.9 397.4c0 2-2.3 3.6-5.2 3.6-3.3 .3-5.6-1.3-5.6-3.6 0-2 2.3-3.6 5.2-3.6 3-.3 5.6 1.3 5.6 3.6zm-31.1-4.5c-.7 2 1.3 4.3 4.3 4.9 2.6 1 5.6 0 6.2-2s-1.3-4.3-4.3-5.2c-2.6-.7-5.5 .3-6.2 2.3zm44.2-1.7c-2.9 .7-4.9 2.6-4.6 4.9 .3 2 2.9 3.3 5.9 2.6 2.9-.7 4.9-2.6 4.6-4.6-.3-1.9-3-3.2-5.9-2.9zM244.8 8C106.1 8 0 113.3 0 252c0 110.9 69.8 205.8 169.5 239.2 12.8 2.3 17.3-5.6 17.3-12.1 0-6.2-.3-40.4-.3-61.4 0 0-70 15-84.7-29.8 0 0-11.4-29.1-27.8-36.6 0 0-22.9-15.7 1.6-15.4 0 0 24.9 2 38.6 25.8 21.9 38.6 58.6 27.5 72.9 20.9 2.3-16 8.8-27.1 16-33.7-55.9-6.2-112.3-14.3-112.3-110.5 0-27.5 7.6-41.3 23.6-58.9-2.6-6.5-11.1-33.3 2.6-67.9 20.9-6.5 69 27 69 27 20-5.6 41.5-8.5 62.8-8.5s42.8 2.9 62.8 8.5c0 0 48.1-33.6 69-27 13.7 34.7 5.2 61.4 2.6 67.9 16 17.7 25.8 31.5 25.8 58.9 0 96.5-58.9 104.2-114.8 110.5 9.2 7.9 17 22.9 17 46.4 0 33.7-.3 75.4-.3 83.6 0 6.5 4.6 14.4 17.3 12.1C428.2 457.8 496 362.9 496 252 496 113.3 383.5 8 244.8 8zM97.2 352.9c-1.3 1-1 3.3 .7 5.2 1.6 1.6 3.9 2.3 5.2 1 1.3-1 1-3.3-.7-5.2-1.6-1.6-3.9-2.3-5.2-1zm-10.8-8.1c-.7 1.3 .3 2.9 2.3 3.9 1.6 1 3.6 .7 4.3-.7 .7-1.3-.3-2.9-2.3-3.9-2-.6-3.6-.3-4.3 .7zm32.4 35.6c-1.6 1.3-1 4.3 1.3 6.2 2.3 2.3 5.2 2.6 6.5 1 1.3-1.3 .7-4.3-1.3-6.2-2.2-2.3-5.2-2.6-6.5-1zm-11.4-14.7c-1.6 1-1.6 3.6 0 5.9 1.6 2.3 4.3 3.3 5.6 2.3 1.6-1.3 1.6-3.9 0-6.2-1.4-2.3-4-3.3-5.6-2z\"/></svg>\n",
-    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/canary/docs/guides/train-arrhythmia-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/main/docs/guides/train-arrhythmia-model.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    GitHub source\n",
     "</a>\n",
     "\n",
diff --git a/notebooks/train-ecg-denoiser.ipynb b/notebooks/train-ecg-denoiser.ipynb
index 71de1d5e..b450b12d 100644
--- a/notebooks/train-ecg-denoiser.ipynb
+++ b/notebooks/train-ecg-denoiser.ipynb
@@ -8,12 +8,12 @@
     "<div class=\"grid cards\" markdown>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 488 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M488 261.8C488 403.3 391.1 504 248 504 110.8 504 0 393.2 0 256S110.8 8 248 8c66.8 0 123 24.5 166.3 64.9l-67.5 64.9C258.5 52.6 94.3 116.6 94.3 256c0 86.5 69.1 156.6 153.7 156.6 98.2 0 135-70.4 140.8-106.9H248v-85.3h236.1c2.3 12.7 3.9 24.9 3.9 41.4z\"/></svg>\n",
-    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/canary/docs/guides/train-ecg-denoiser.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/main/docs/guides/train-ecg-denoiser.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    View in Colab\n",
     "</a>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 496 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M165.9 397.4c0 2-2.3 3.6-5.2 3.6-3.3 .3-5.6-1.3-5.6-3.6 0-2 2.3-3.6 5.2-3.6 3-.3 5.6 1.3 5.6 3.6zm-31.1-4.5c-.7 2 1.3 4.3 4.3 4.9 2.6 1 5.6 0 6.2-2s-1.3-4.3-4.3-5.2c-2.6-.7-5.5 .3-6.2 2.3zm44.2-1.7c-2.9 .7-4.9 2.6-4.6 4.9 .3 2 2.9 3.3 5.9 2.6 2.9-.7 4.9-2.6 4.6-4.6-.3-1.9-3-3.2-5.9-2.9zM244.8 8C106.1 8 0 113.3 0 252c0 110.9 69.8 205.8 169.5 239.2 12.8 2.3 17.3-5.6 17.3-12.1 0-6.2-.3-40.4-.3-61.4 0 0-70 15-84.7-29.8 0 0-11.4-29.1-27.8-36.6 0 0-22.9-15.7 1.6-15.4 0 0 24.9 2 38.6 25.8 21.9 38.6 58.6 27.5 72.9 20.9 2.3-16 8.8-27.1 16-33.7-55.9-6.2-112.3-14.3-112.3-110.5 0-27.5 7.6-41.3 23.6-58.9-2.6-6.5-11.1-33.3 2.6-67.9 20.9-6.5 69 27 69 27 20-5.6 41.5-8.5 62.8-8.5s42.8 2.9 62.8 8.5c0 0 48.1-33.6 69-27 13.7 34.7 5.2 61.4 2.6 67.9 16 17.7 25.8 31.5 25.8 58.9 0 96.5-58.9 104.2-114.8 110.5 9.2 7.9 17 22.9 17 46.4 0 33.7-.3 75.4-.3 83.6 0 6.5 4.6 14.4 17.3 12.1C428.2 457.8 496 362.9 496 252 496 113.3 383.5 8 244.8 8zM97.2 352.9c-1.3 1-1 3.3 .7 5.2 1.6 1.6 3.9 2.3 5.2 1 1.3-1 1-3.3-.7-5.2-1.6-1.6-3.9-2.3-5.2-1zm-10.8-8.1c-.7 1.3 .3 2.9 2.3 3.9 1.6 1 3.6 .7 4.3-.7 .7-1.3-.3-2.9-2.3-3.9-2-.6-3.6-.3-4.3 .7zm32.4 35.6c-1.6 1.3-1 4.3 1.3 6.2 2.3 2.3 5.2 2.6 6.5 1 1.3-1.3 .7-4.3-1.3-6.2-2.2-2.3-5.2-2.6-6.5-1zm-11.4-14.7c-1.6 1-1.6 3.6 0 5.9 1.6 2.3 4.3 3.3 5.6 2.3 1.6-1.3 1.6-3.9 0-6.2-1.4-2.3-4-3.3-5.6-2z\"/></svg>\n",
-    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/canary/docs/guides/train-ecg-denoiser.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/main/docs/guides/train-ecg-denoiser.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    GitHub source\n",
     "</a>\n",
     "\n",
diff --git a/notebooks/train-ecg-segmentation.ipynb b/notebooks/train-ecg-segmentation.ipynb
index 7ed6ba14..e688dba0 100644
--- a/notebooks/train-ecg-segmentation.ipynb
+++ b/notebooks/train-ecg-segmentation.ipynb
@@ -8,12 +8,12 @@
     "<div class=\"grid cards\" markdown>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 488 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M488 261.8C488 403.3 391.1 504 248 504 110.8 504 0 393.2 0 256S110.8 8 248 8c66.8 0 123 24.5 166.3 64.9l-67.5 64.9C258.5 52.6 94.3 116.6 94.3 256c0 86.5 69.1 156.6 153.7 156.6 98.2 0 135-70.4 140.8-106.9H248v-85.3h236.1c2.3 12.7 3.9 24.9 3.9 41.4z\"/></svg>\n",
-    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/canary/docs/guides/train-ecg-segmentation.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://colab.research.google.com/github/AmbiqAI/heartkit/blob/main/docs/guides/train-ecg-segmentation.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    View in Colab\n",
     "</a>\n",
     "\n",
     "- <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"50\" width=\"50\" viewBox=\"0 0 496 512\"><!--!Font Awesome Free 6.6.0 by @fontawesome - https://fontawesome.com License - https://fontawesome.com/license/free Copyright 2024 Fonticons, Inc.--><path fill=\"#2094F3\" d=\"M165.9 397.4c0 2-2.3 3.6-5.2 3.6-3.3 .3-5.6-1.3-5.6-3.6 0-2 2.3-3.6 5.2-3.6 3-.3 5.6 1.3 5.6 3.6zm-31.1-4.5c-.7 2 1.3 4.3 4.3 4.9 2.6 1 5.6 0 6.2-2s-1.3-4.3-4.3-5.2c-2.6-.7-5.5 .3-6.2 2.3zm44.2-1.7c-2.9 .7-4.9 2.6-4.6 4.9 .3 2 2.9 3.3 5.9 2.6 2.9-.7 4.9-2.6 4.6-4.6-.3-1.9-3-3.2-5.9-2.9zM244.8 8C106.1 8 0 113.3 0 252c0 110.9 69.8 205.8 169.5 239.2 12.8 2.3 17.3-5.6 17.3-12.1 0-6.2-.3-40.4-.3-61.4 0 0-70 15-84.7-29.8 0 0-11.4-29.1-27.8-36.6 0 0-22.9-15.7 1.6-15.4 0 0 24.9 2 38.6 25.8 21.9 38.6 58.6 27.5 72.9 20.9 2.3-16 8.8-27.1 16-33.7-55.9-6.2-112.3-14.3-112.3-110.5 0-27.5 7.6-41.3 23.6-58.9-2.6-6.5-11.1-33.3 2.6-67.9 20.9-6.5 69 27 69 27 20-5.6 41.5-8.5 62.8-8.5s42.8 2.9 62.8 8.5c0 0 48.1-33.6 69-27 13.7 34.7 5.2 61.4 2.6 67.9 16 17.7 25.8 31.5 25.8 58.9 0 96.5-58.9 104.2-114.8 110.5 9.2 7.9 17 22.9 17 46.4 0 33.7-.3 75.4-.3 83.6 0 6.5 4.6 14.4 17.3 12.1C428.2 457.8 496 362.9 496 252 496 113.3 383.5 8 244.8 8zM97.2 352.9c-1.3 1-1 3.3 .7 5.2 1.6 1.6 3.9 2.3 5.2 1 1.3-1 1-3.3-.7-5.2-1.6-1.6-3.9-2.3-5.2-1zm-10.8-8.1c-.7 1.3 .3 2.9 2.3 3.9 1.6 1 3.6 .7 4.3-.7 .7-1.3-.3-2.9-2.3-3.9-2-.6-3.6-.3-4.3 .7zm32.4 35.6c-1.6 1.3-1 4.3 1.3 6.2 2.3 2.3 5.2 2.6 6.5 1 1.3-1.3 .7-4.3-1.3-6.2-2.2-2.3-5.2-2.6-6.5-1zm-11.4-14.7c-1.6 1-1.6 3.6 0 5.9 1.6 2.3 4.3 3.3 5.6 2.3 1.6-1.3 1.6-3.9 0-6.2-1.4-2.3-4-3.3-5.6-2z\"/></svg>\n",
-    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/canary/docs/guides/train-ecg-segmentation.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
+    "<a href=\"https://github.com/AmbiqAI/heartkit/blob/main/docs/guides/train-ecg-segmentation.ipynb\" class=\"md-content__button md-icon\" style=\"color: #2094F3;\">\n",
     "    GitHub source\n",
     "</a>\n",
     "\n",
diff --git a/poetry.lock b/poetry.lock
index d8fe5c8e..925acf99 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -2520,32 +2520,28 @@ files = [
 
 [[package]]
 name = "neuralspot-edge"
-version = "0.1.3"
+version = "0.1.5"
 description = ""
 optional = false
-python-versions = ">=3.11,<3.13"
-files = []
-develop = false
+python-versions = "<3.13,>=3.10"
+files = [
+    {file = "neuralspot_edge-0.1.5-py3-none-any.whl", hash = "sha256:6be9f01fbf97987b082fe452bd8e80ebc8d57a2fce6edd2f3443afaf6e21da61"},
+    {file = "neuralspot_edge-0.1.5.tar.gz", hash = "sha256:598dc8822f99d19b2910a54d75f490c820e1d5cc8b7b0bdc689f02cf400de0cd"},
+]
 
 [package.dependencies]
-boto3 = "^1.34.151"
-h5py = "^3.10.0"
-keras = "^3.0.4"
-matplotlib = "^3.9.0"
-pandas = "^2.2.2"
-plotly = "^5.22.0"
-pydantic = "^2.6.1"
-requests = "^2.31.0"
-scikit-learn = "^1.5.1"
-seaborn = "^0.13.2"
-tensorflow = "^2.16.1"
-tqdm = "^4.66.4"
-
-[package.source]
-type = "git"
-url = "https://github.com/AmbiqAI/neuralspot-edge.git"
-reference = "HEAD"
-resolved_reference = "b5ed7f5d884d1297728c94a39eef1ba9cff3fe07"
+boto3 = ">=1.34.151,<2.0.0"
+h5py = ">=3.10.0,<4.0.0"
+keras = ">=3.0.4,<4.0.0"
+matplotlib = ">=3.9.0,<4.0.0"
+pandas = ">=2.2.2,<3.0.0"
+plotly = ">=5.22.0,<6.0.0"
+pydantic = ">=2.6.1,<3.0.0"
+requests = ">=2.31.0,<3.0.0"
+scikit-learn = ">=1.5.1,<2.0.0"
+seaborn = ">=0.13.2,<0.14.0"
+tensorflow = ">=2.16.1,<3.0.0"
+tqdm = ">=4.66.4,<5.0.0"
 
 [[package]]
 name = "nodeenv"
@@ -5180,4 +5176,4 @@ files = [
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.11,<3.13"
-content-hash = "1a3b60841bfb644cf9426a21237665cca6872a7a0963d7b2f8dcd755dc0ae3ed"
+content-hash = "7a84e8e0e769fd400676192f5773bad99e6d9bf8a20a176971ed5aada44983fa"
diff --git a/pyproject.toml b/pyproject.toml
index dc98ed9a..4e087aea 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -58,7 +58,7 @@ orjson = "^3.9.13"
 physiokit = "^0.8.1"
 requests = "^2.31.0"
 argdantic = {extras = ["all"], version = "^1.0.0"}
-neuralspot-edge = {git = "https://github.com/AmbiqAI/neuralspot-edge.git"}
+neuralspot-edge = "^0.1.5"
 
 [tool.poetry.group.dev.dependencies]
 ipython = "^8.21.0"
diff --git a/scripts/upload_to_model_zoo.py b/scripts/upload_to_model_zoo.py
index 4c6ca1be..1d3e8405 100644
--- a/scripts/upload_to_model_zoo.py
+++ b/scripts/upload_to_model_zoo.py
@@ -78,18 +78,19 @@ def upload_to_model_zoo(
     parser()
 
 """
-python ./scripts/upload_to_model_zoo.py -t rhythm -s ./results/arr-2-eff-sm -n arr-2-eff-sm -v latest
-python ./scripts/upload_to_model_zoo.py -t rhyhtm -s ./results/arr-4-eff-sm -n arr-4-eff-sm -v latest
+python ./scripts/upload_to_model_zoo.py -t rhythm -s ./results/arr-2-eff-sm -n arr-2-eff-sm -v v1.0
+python ./scripts/upload_to_model_zoo.py -t rhyhtm -s ./results/arr-4-eff-sm -n arr-4-eff-sm -v v1.0
 
-python ./scripts/upload_to_model_zoo.py -t beat -s ./results/beat-2-eff-sm -n beat-2-eff-sm -v latest
-python ./scripts/upload_to_model_zoo.py -t beat -s ./results/beat-3-eff-sm -n beat-3-eff-sm -v latest
+python ./scripts/upload_to_model_zoo.py -t beat -s ./results/beat-2-eff-sm -n beat-2-eff-sm -v v1.0
+python ./scripts/upload_to_model_zoo.py -t beat -s ./results/beat-3-eff-sm -n beat-3-eff-sm -v v1.0
 
-python ./scripts/upload_to_model_zoo.py -t denoise -s ./results/den-ppg-tcn-sm -n den-ppg-tcn-sm -v latest
-python ./scripts/upload_to_model_zoo.py -t denoise -s ./results/den-tcn-sm -n den-tcn-sm -v latest
+python ./scripts/upload_to_model_zoo.py -t denoise -s ./results/den-ppg-tcn-sm -n den-ppg-tcn-sm -v v1.0
+python ./scripts/upload_to_model_zoo.py -t denoise -s ./results/den-tcn-sm -n den-tcn-sm -v v1.0
 
-python ./scripts/upload_to_model_zoo.py -t foundation -s ./results/fnd-eff-sm -n fnd-eff-sm -v latest
+python ./scripts/upload_to_model_zoo.py -t foundation -s ./results/fnd-eff-sm -n fnd-eff-sm -v v1.0
 
-python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-2-tcn-sm -n seg-2-tcn-sm -v latest
-python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-4-tcn-sm -n seg-4-tcn-sm -v latest
-python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-ppg-2-tcn-sm -n seg-ppg-2-tcn-sm -v latest
+python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-2-tcn-sm -n seg-2-tcn-sm -v v1.0
+python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-4-tcn-sm -n seg-4-tcn-sm -v v1.0
+python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-4-tcn-lg -n seg-4-tcn-lg -v v1.0
+python ./scripts/upload_to_model_zoo.py -t segmentation -s ./results/seg-ppg-2-tcn-sm -n seg-ppg-2-tcn-sm -v v1.0
 """