deepxde.nn.jax
==============

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: