diff --git a/mkdocs.yml b/mkdocs.yml
index 2b7c9f1..6c338f4 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -7,35 +7,18 @@ repo_name: flaport/sax
 copyright: Copyright © 2025, Floris Laporte, Apache-2.0
 
 theme:
-  name: material
+  name: shadcn
   logo: assets/logo.svg
   favicon: assets/favicon.ico
-  font:
-    text: Roboto
-    code: Roboto Mono
-  palette:
-    - media: "(prefers-color-scheme)"
-      toggle:
-        icon: material/brightness-auto
-        name: Switch to light mode
-    - media: "(prefers-color-scheme: light)"
-      scheme: default
-      toggle:
-        icon: material/weather-sunny
-        name: Switch to dark mode
-      primary: blue
-      accent: deep purple
-    - media: "(prefers-color-scheme: dark)"
-      scheme: slate
-      toggle:
-        icon: material/weather-night
-        name: Switch to system default
-      primary: blue
-      accent: deep purple
-  features:
-    - content.code.copy
-    - navigation.footer
-  search: true
+  pygments_style:
+    light: shadcn-light
+    dark: github-dark
+  katex_options: {}
+  crossref_modules:
+    - sax
+    - sax.fit
+    - sax.models
+    - sax.models.rf
 
 nav:
   - home: index.md
@@ -65,12 +48,6 @@ nav:
       - sax.models.rf: models_rf.md
   - changelog: changelog.md
 
-extra_css:
-  - assets/custom.css
-
-extra_javascript:
-  - https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js
-
 plugins:
   - search
   - autorefs
@@ -118,7 +95,8 @@ markdown_extensions:
   - footnotes
   - toc:
         permalink: true
-  - pymdownx.emoji
+  - pymdownx.arithmatex:
+      generic: true
   - pymdownx.highlight:
       anchor_linenums: true
       line_spans: __span
@@ -128,16 +106,10 @@ markdown_extensions:
   - pymdownx.magiclink
   - pymdownx.tabbed:
       alternate_style: true
+  - pymdownx.superfences
   - admonition
   - pymdownx.details
-  - pymdownx.arithmatex:
-      generic: true
-  - pymdownx.superfences
+  - pymdownx.emoji
 
 hooks:
   - docs/hooks.py
-
-extra:
-  social:
-    - icon: fontawesome/brands/github
-      link: https://github.com/flaport
diff --git a/pyproject.toml b/pyproject.toml
index 6e6f561..6a3908d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,7 +22,7 @@ docs = [
   "mkautodoc>=0.2.0,<0.3.0",
   "mkdocs-autorefs>=1.4.4,<1.5.0",
   "mkdocs-bibtex>=4.4.0,<5.0.0",
-  "mkdocs-material>=9.7.6,<10.0.0",
+  "mkdocs-shadcn @ git+https://github.com/gdsfactory/mkdocs-shadcn.git",
   "mkdocs-matplotlib>=0.10.1,<0.11.0",
   "mkdocs>=1.6.1,<2.0.0",
   "mkdocstrings[python]>=1.0.4,<2.0.0",