Update JAX nn docs

lululxvi · Feb 23, 2022 · a0e072f · a0e072f
1 parent 6b1c737
commit a0e072f
Showing 1 changed file with 2 additions and 13 deletions.
diff --git a/docs/modules/deepxde.nn.jax.rst b/docs/modules/deepxde.nn.jax.rst
@@ -1,8 +1,5 @@
-deepxde.nn.jax package
-======================
-
-Submodules
-----------
+deepxde.nn.jax
+==============
 
 deepxde.nn.jax.fnn module
 -------------------------
@@ -19,11 +16,3 @@ deepxde.nn.jax.nn module
    :members:
    :undoc-members:
    :show-inheritance:
-
-Module contents
----------------
-
-.. automodule:: deepxde.nn.jax
-   :members:
-   :undoc-members:
-   :show-inheritance: