zea.backend.jax
===============

.. automodule:: zea.backend.jax

   
   .. rubric:: Functions

   .. autosummary::
   
      on_device_jax