Update docs/modules.rst to include missing modules

Author: google-labs-jules[bot]

docs/images/example1.jpg

@@ -38,9 +38,6 @@ layers
    :undoc-members:
    :show-inheritance:
 
-
-
-
 sparse
 ------
 
@@ -64,3 +61,51 @@ shtlc
    :members:
    :undoc-members:
    :show-inheritance:
+
+coupling
+--------
+
+.. automodule:: vbjax.coupling
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+diagnostics
+-----------
+
+.. automodule:: vbjax.diagnostics
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+embed
+-----
+
+.. automodule:: vbjax.embed
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+implicit
+--------
+
+.. automodule:: vbjax.implicit
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+regmap
+------
+
+.. automodule:: vbjax.regmap
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+signal
+------
+
+.. automodule:: vbjax.signal
+   :members:
+   :undoc-members:
+   :show-inheritance: