Update API

lululxvi · lululxvi · commit 768b02258e1e · 2022-02-23T15:21:57.000-05:00
diff --git a/docs/modules/deepxde.backend.jax.rst b/docs/modules/deepxde.backend.jax.rst
@@ -0,0 +1,21 @@
+deepxde.backend.jax package
+===========================
+
+Submodules
+----------
+
+deepxde.backend.jax.tensor module
+---------------------------------
+
+.. automodule:: deepxde.backend.jax.tensor
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+Module contents
+---------------
+
+.. automodule:: deepxde.backend.jax
+   :members:
+   :undoc-members:
+   :show-inheritance:
diff --git a/docs/modules/deepxde.backend.rst b/docs/modules/deepxde.backend.rst
@@ -7,6 +7,7 @@ Subpackages
 .. toctree::
    :maxdepth: 4
 
+   deepxde.backend.jax
    deepxde.backend.pytorch
    deepxde.backend.tensorflow
    deepxde.backend.tensorflow_compat_v1
diff --git a/docs/modules/deepxde.data.rst b/docs/modules/deepxde.data.rst
@@ -81,6 +81,14 @@ deepxde.data.pde module
    :undoc-members:
    :show-inheritance:
 
+deepxde.data.quadruple module
+-----------------------------
+
+.. automodule:: deepxde.data.quadruple
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
 deepxde.data.sampler module
 ---------------------------
 
diff --git a/docs/modules/deepxde.icbc.rst b/docs/modules/deepxde.icbc.rst
@@ -0,0 +1,19 @@
+deepxde.icbc
+============
+
+deepxde.icbc.boundary\_conditions module
+----------------------------------------
+
+.. automodule:: deepxde.icbc.boundary_conditions
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+deepxde.icbc.initial\_conditions module
+---------------------------------------
+
+.. automodule:: deepxde.icbc.initial_conditions
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
diff --git a/docs/modules/deepxde.icbcs.rst b/docs/modules/deepxde.icbcs.rst
diff --git a/docs/modules/deepxde.nn.jax.rst b/docs/modules/deepxde.nn.jax.rst
@@ -0,0 +1,29 @@
+deepxde.nn.jax package
+======================
+
+Submodules
+----------
+
+deepxde.nn.jax.fnn module
+-------------------------
+
+.. automodule:: deepxde.nn.jax.fnn
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+deepxde.nn.jax.nn module
+------------------------
+
+.. automodule:: deepxde.nn.jax.nn
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+Module contents
+---------------
+
+.. automodule:: deepxde.nn.jax
+   :members:
+   :undoc-members:
+   :show-inheritance:
diff --git a/docs/modules/deepxde.nn.tensorflow_compat_v1.rst b/docs/modules/deepxde.nn.tensorflow_compat_v1.rst
@@ -25,6 +25,14 @@ deepxde.nn.tensorflow\_compat\_v1.mfnn module
    :undoc-members:
    :show-inheritance:
 
+deepxde.nn.tensorflow\_compat\_v1.mionet module
+-----------------------------------------------
+
+.. automodule:: deepxde.nn.tensorflow_compat_v1.mionet
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
 deepxde.nn.tensorflow\_compat\_v1.msffn module
 ----------------------------------------------
 
diff --git a/docs/modules/deepxde.optimizers.jax.rst b/docs/modules/deepxde.optimizers.jax.rst
@@ -0,0 +1,21 @@
+deepxde.optimizers.jax package
+==============================
+
+Submodules
+----------
+
+deepxde.optimizers.jax.optimizers module
+----------------------------------------
+
+.. automodule:: deepxde.optimizers.jax.optimizers
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+Module contents
+---------------
+
+.. automodule:: deepxde.optimizers.jax
+   :members:
+   :undoc-members:
+   :show-inheritance:
