1 files changed, 26 insertions, 78 deletions
diff --git a/docs/source/all.rst b/docs/source/all.rst
index c968aa1..d7b878f 100644
--- a/docs/source/all.rst
+++ b/docs/source/all.rst
@@ -1,88 +1,36 @@
 
+.. _sphx_glr_api_reference:
 
-Python modules
-==============
+API and modules
+===============
 
-ot
---
+.. currentmodule:: ot
 
-.. automodule:: ot
-   :members:
-
-ot.lp
------
-.. automodule:: ot.lp
-   :members:
-
-ot.bregman
-----------
-
-.. automodule:: ot.bregman
-   :members:
-
-ot.smooth
------
-.. automodule:: ot.smooth
-   :members:
-
-ot.gromov
-----------
-
-.. automodule:: ot.gromov
-   :members:
-
-
-ot.optim
---------
-
-.. automodule:: ot.optim
-   :members:
 
-ot.da
---------
+:py:mod:`ot`:
 
-.. automodule:: ot.da
-  :members:
+.. autosummary::
+   :toctree: gen_modules/
+   :template: module.rst
 
-ot.gpu
---------
+   lp
+   bregman
+   smooth
+   gromov
+   optim
+   da
+   gpu
+   dr
+   utils
+   datasets
+   plot
+   stochastic
+   unbalanced
+   partial
 
-.. automodule:: ot.gpu
-  :members:
+.. autosummary::
+   :toctree: ../modules/generated/
+   :template: module.rst
 
-ot.dr
---------
-
-.. automodule:: ot.dr
-  :members:
-
-
-ot.utils
---------
-
-.. automodule:: ot.utils
-   :members:
-
-ot.datasets
------------
-
-.. automodule:: ot.datasets
-   :members:
-
-ot.plot
--------
-
-.. automodule:: ot.plot
-   :members:
-
-ot.stochastic
--------------
-
-.. automodule:: ot.stochastic
+.. automodule:: ot
    :members:
-
-ot.unbalanced
--------------
-
-.. automodule:: ot.unbalanced
-  :members: