diff --git a/docs/javascripts/katex.js b/docs/javascripts/katex.js
new file mode 100644
index 0000000..956b86a
--- /dev/null
+++ b/docs/javascripts/katex.js
@@ -0,0 +1,16 @@
+const renderMath = (el) => {
+  renderMathInElement(el, {
+    delimiters: [
+      { left: "$$",  right: "$$",  display: true },
+      { left: "$",   right: "$",   display: false },
+      { left: "\\(", right: "\\)", display: false },
+      { left: "\\[", right: "\\]", display: true }
+    ],
+  });
+};
+
+if (typeof document$ !== "undefined") {
+  document$.subscribe(({ body }) => renderMath(body));
+} else {
+  document.addEventListener("DOMContentLoaded", () => renderMath(document.body));
+}
diff --git a/docs/javascripts/mathjax.js b/docs/javascripts/mathjax.js
deleted file mode 100644
index f5e96e7..0000000
--- a/docs/javascripts/mathjax.js
+++ /dev/null
@@ -1,18 +0,0 @@
-window.MathJax = {
-  tex: {
-    inlineMath: [["\\(", "\\)"]],
-    displayMath: [["\\[", "\\]"]],
-    processEscapes: true,
-    processEnvironments: true
-  },
-  options: {
-    ignoreHtmlClass: ".*|",
-    processHtmlClass: "arithmatex"
-  }
-};
-
-document.addEventListener("DOMContentLoaded", () => {
-  MathJax.startup.document.state(0);
-  MathJax.typesetClear();
-  MathJax.typesetPromise();
-});
diff --git a/mkdocs.yml b/mkdocs.yml
index 3084148..7e6c856 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -11,6 +11,10 @@ markdown_extensions:
   - pymdownx.arithmatex:
       generic: true
 
+extra_css:
+  - https://unpkg.com/katex@0/dist/katex.min.css
+
 extra_javascript:
-  - javascripts/mathjax.js
-  - https://unpkg.com/mathjax@3/es5/tex-mml-chtml.js
+  - javascripts/katex.js
+  - https://unpkg.com/katex@0/dist/katex.min.js
+  - https://unpkg.com/katex@0/dist/contrib/auto-render.min.js