diff --git a/docs/javascripts/mathjax.js b/docs/javascripts/mathjax.js
new file mode 100644
index 00000000..daf3bc42
--- /dev/null
+++ b/docs/javascripts/mathjax.js
@@ -0,0 +1,19 @@
+window.MathJax = {
+  tex: {
+    inlineMath: [["\\(", "\\)"]],
+    displayMath: [["\\[", "\\]"]],
+    processEscapes: true,
+    processEnvironments: true
+  },
+  options: {
+    ignoreHtmlClass: ".*|",
+    processHtmlClass: "arithmatex"
+  }
+};
+
+document$.subscribe(() => {
+  MathJax.startup.output.clearCache();
+  MathJax.typesetClear();
+  MathJax.texReset();
+  MathJax.typesetPromise();
+});
diff --git a/mkdocs.yml b/mkdocs.yml
index 2d8e168c..80168724 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -72,5 +72,6 @@ markdown_extensions:
   - md_in_html
 
 extra_javascript:
+  - javascripts/mathjax.js
   - https://polyfill.io/v3/polyfill.min.js?features=es6
   - https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js
diff --git a/pyproject.toml b/pyproject.toml
index 9319c253..2087d738 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -21,6 +21,7 @@ dependencies = [
     "hdf5plugin>=7.0.0",
     "matplotlib>=3.8",
     "seaborn>=0.13.2",
+    "tlviz>=0.1.1",
 ]
 
 readme = "README.md"
diff --git a/uv.lock b/uv.lock
index f0f98e6c..32cc3b0c 100644
--- a/uv.lock
+++ b/uv.lock
@@ -2108,6 +2108,7 @@ dependencies = [
     { name = "scipy" },
     { name = "seaborn" },
     { name = "tensorly" },
+    { name = "tlviz" },
     { name = "tqdm" },
     { name = "vsparse" },
 ]
@@ -2174,6 +2175,7 @@ requires-dist = [
     { name = "scipy", specifier = ">=1.16" },
     { name = "seaborn", specifier = ">=0.13.2" },
     { name = "tensorly", specifier = ">=0.9.0" },
+    { name = "tlviz", specifier = ">=0.1.1" },
     { name = "tqdm", specifier = ">=4.66.1" },
     { name = "vsparse", specifier = ">=0.2.0" },
 ]