dymad.utils.wrapper
===================

.. automodule:: dymad.utils.wrapper

   
   .. rubric:: Functions

   .. autosummary::
   
      jax_to_torch
      torch_to_jax
   
   .. rubric:: Classes

   .. autosummary::
   
      JaxMultiInFn
      JaxWrapper