waxmorph.jax
============
.. rubric:: Modules

.. autosummary::
   :toctree:
   :recursive:

   waxmorph.jax.gnn
   waxmorph.jax.graph
   waxmorph.jax.losses
   waxmorph.jax.mlp
   waxmorph.jax.warp_autograd


.. currentmodule:: waxmorph.jax

.. automodule:: waxmorph.jax