waxmorph.jax.warp\_autograd
===========================

.. currentmodule:: waxmorph.jax.warp_autograd

.. automodule:: waxmorph.jax.warp_autograd

   
   .. rubric:: Functions

   .. autosummary::
      :toctree:
   
      warp_diffusion_step
      warp_mech_step