``braincore.transform`` module
==============================

.. currentmodule:: braincore.transform 
.. automodule:: braincore.transform 



Transform Functions
-------------------

.. autosummary::
   :toctree: generated/

   vector_grad
   grad
   jacrev
   jacfwd
   jacobian
   hessian
   cond
   switch
   ifelse
   scan
   for_loop
   while_loop
   jit


Transform Tools
---------------

.. autosummary::
   :toctree: generated/
   :nosignatures:
   :template: classtemplate.rst

   StatefulFunction
   ProgressBar
   make_jaxpr
   jit_error