Update docs

lululxvi · lululxvi · commit 6a067667a6a5 · 2025-01-03T20:30:56.000-05:00
diff --git a/docs/index.rst b/docs/index.rst
@@ -107,6 +107,7 @@ If you are looking for information on a specific function, class or method, this
   modules/deepxde.nn.tensorflow
   modules/deepxde.nn.tensorflow_compat_v1
   modules/deepxde.optimizers
+  modules/deepxde.optimizers.pytorch
   modules/deepxde.utils
 
 Indices and tables
diff --git a/docs/modules/deepxde.optimizers.pytorch.rst b/docs/modules/deepxde.optimizers.pytorch.rst
@@ -1,21 +1,11 @@
-deepxde.optimizers.pytorch package
-==================================
+deepxde.optimizers.pytorch
+==========================
 
-Submodules
-----------
+deepxde.optimizers.pytorch.nncg module
+--------------------------------------
 
-deepxde.optimizers.pytorch.optimizers module
---------------------------------------------
-
-.. automodule:: deepxde.optimizers.pytorch.optimizers
+.. automodule:: deepxde.optimizers.pytorch.nncg
    :members:
    :undoc-members:
    :show-inheritance:
 
-Module contents
----------------
-
-.. automodule:: deepxde.optimizers.pytorch
-   :members:
-   :undoc-members:
-   :show-inheritance:
