diff --git a/docs/modules/data_indices.rst b/docs/modules/data_indices.rst
new file mode 100644
index 0000000..d776aab
--- /dev/null
+++ b/docs/modules/data_indices.rst
@@ -0,0 +1,8 @@
+##############
+ Data Indices
+##############
+
+.. automodule:: anemoi.models.data_indices
+   :members:
+   :no-undoc-members:
+   :show-inheritance:
diff --git a/docs/modules/distributed.rst b/docs/modules/distributed.rst
new file mode 100644
index 0000000..6aa8811
--- /dev/null
+++ b/docs/modules/distributed.rst
@@ -0,0 +1,8 @@
+#############
+ Distributed
+#############
+
+.. automodule:: anemoi.models.distributed
+   :members:
+   :no-undoc-members:
+   :show-inheritance:
diff --git a/docs/modules/interface.rst b/docs/modules/interface.rst
new file mode 100644
index 0000000..28e041f
--- /dev/null
+++ b/docs/modules/interface.rst
@@ -0,0 +1,8 @@
+###########
+ Interface
+###########
+
+.. automodule:: anemoi.models.interface
+   :members:
+   :no-undoc-members:
+   :show-inheritance:
diff --git a/docs/modules/layers.rst b/docs/modules/layers.rst
new file mode 100644
index 0000000..d93d2c7
--- /dev/null
+++ b/docs/modules/layers.rst
@@ -0,0 +1,8 @@
+########
+ Layers
+########
+
+.. automodule:: anemoi.models.layers
+   :members:
+   :no-undoc-members:
+   :show-inheritance:
diff --git a/docs/modules/models.rst b/docs/modules/models.rst
new file mode 100644
index 0000000..745046b
--- /dev/null
+++ b/docs/modules/models.rst
@@ -0,0 +1,8 @@
+########
+ Models
+########
+
+.. automodule:: anemoi.models.models
+   :members:
+   :no-undoc-members:
+   :show-inheritance:
diff --git a/docs/modules/preprocessing.rst b/docs/modules/preprocessing.rst
new file mode 100644
index 0000000..bccb0d1
--- /dev/null
+++ b/docs/modules/preprocessing.rst
@@ -0,0 +1,8 @@
+###############
+ Preprocessing
+###############
+
+.. automodule:: anemoi.models.preprocessing
+   :members:
+   :no-undoc-members:
+   :show-inheritance:
diff --git a/docs/modules/utils.rst b/docs/modules/utils.rst
new file mode 100644
index 0000000..daf5133
--- /dev/null
+++ b/docs/modules/utils.rst
@@ -0,0 +1,8 @@
+#######
+ Utils
+#######
+
+.. automodule:: anemoi.models.utils
+   :members:
+   :no-undoc-members:
+   :show-inheritance: