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

.. automodule:: zea.backend

   
   .. rubric:: Functions

   .. autosummary::
   
      func_on_device
      jit
      tf_function
   
   .. rubric:: Classes

   .. autosummary::
   
      on_device
   
.. rubric:: Modules

.. autosummary::
   :toctree:
   :recursive:

   autograd
   jax
   optimizer
   tensorflow
   torch