diff --git a/.github/workflows/ci_docs.yml b/.github/workflows/ci_docs.yml
index d93052c04..fc2fa37a4 100644
--- a/.github/workflows/ci_docs.yml
+++ b/.github/workflows/ci_docs.yml
@@ -39,8 +39,7 @@ jobs:
       - name: Install Texlive & tree
         run: |
           sudo apt-get update --fix-missing
-          # install Texlive, see https://linuxconfig.org/how-to-install-latex-on-ubuntu-20-04-focal-fossa-linux
-          sudo apt-get install -y cmake tree pandoc texlive-latex-extra dvipng texlive-pictures
+          sudo apt-get install -y cmake tree pandoc
 
       - name: Install dependencies
         run: |
diff --git a/.github/workflows/docs-deploy.yml b/.github/workflows/docs-deploy.yml
index d124acffc..7215b1c1e 100644
--- a/.github/workflows/docs-deploy.yml
+++ b/.github/workflows/docs-deploy.yml
@@ -55,8 +55,6 @@ jobs:
           mkdir -p ${PATH_DATASETS}
           sudo apt-get update --fix-missing
           sudo apt-get install -y cmake pandoc
-          # install Texlive, see https://linuxconfig.org/how-to-install-latex-on-ubuntu-20-04-focal-fossa-linux
-          sudo apt-get install -y texlive-latex-extra dvipng texlive-pictures
           pip install -q -r _requirements/docs.txt -f ${TORCH_URL}
           pip list
         shell: bash
diff --git a/_docs/source/conf.py b/_docs/source/conf.py
index 45a0a3161..f37d8ecb2 100644
--- a/_docs/source/conf.py
+++ b/_docs/source/conf.py
@@ -63,12 +63,12 @@
     "sphinx.ext.intersphinx",
     "sphinx.ext.todo",
     "sphinx.ext.napoleon",
-    "sphinx.ext.imgmath",
     "sphinx.ext.githubpages",
     "nbsphinx",
     "myst_parser",
     "sphinx_paramlinks",
     "pt_lightning_sphinx_theme.extensions.lightning",
+    "sphinx.ext.mathjax",
 ]
 
 # Add any paths that contain templates here, relative to this directory.
@@ -205,6 +205,11 @@
     ),
 ]
 
+# MathJax configuration
+mathjax3_config = {
+    "tex": {"packages": {"[+]": ["ams", "newcommand", "configMacros"]}},
+}
+
 # -- Options for Epub output -------------------------------------------------
 
 # Bibliographic Dublin Core info.