diff --git a/RELEASES.md b/RELEASES.md
index b28bfbda7..0f8918cac 100644
--- a/RELEASES.md
+++ b/RELEASES.md
@@ -28,6 +28,7 @@ This new release adds support for sparse cost matrices and a new lazy EMD solver
 - Fix code coverage (PR #791)
 - Fix test of the version of jax in `ot.backend` (PR #794)
 - Reverting the openmp fix on macOS (PR #789) for macOS (PR #797)
+- Align documentation build dependencies and doc extras (PR #801)
 
 
 ## 0.9.6.post1
diff --git a/docs/requirements.txt b/docs/requirements.txt
index 2e060b93c..cf689f205 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -1,7 +1,9 @@
-sphinx_gallery
-sphinx_rtd_theme
+sphinx
+sphinx-rtd-theme
+sphinx-gallery
+sphinxcontrib-jquery
 numpydoc
-memory_profiler
+myst-parser
 pillow
 networkx
-myst-parser
\ No newline at end of file
+memory_profiler
diff --git a/setup.py b/setup.py
index acbe5aed9..0452fcf25 100644
--- a/setup.py
+++ b/setup.py
@@ -105,6 +105,17 @@
         "dr": ["scikit-learn", "pymanopt", "autograd"],
         "gnn": ["torch", "torch_geometric"],
         "plot": ["matplotlib"],
+        "doc": [
+            "sphinx",
+            "sphinx-rtd-theme",
+            "sphinx-gallery",
+            "sphinxcontrib-jquery",
+            "numpydoc",
+            "myst-parser",
+            "pillow",
+            "networkx",
+            "memory_profiler",
+        ],
         "all": [
             "jax",
             "jaxlib",