``brainunit.autograd`` module
=============================

.. currentmodule:: brainunit.autograd 
.. automodule:: brainunit.autograd 


First-order Gradients
----------------------

.. autosummary::
   :toctree: generated/
   :nosignatures:

    vector_grad
    value_and_grad
    grad


Jacobian Matrix
---------------

.. autosummary::
   :toctree: generated/
   :nosignatures:

    jacrev
    jacfwd
    jacobian



Hessian Matrix
--------------

.. autosummary::
   :toctree: generated/
   :nosignatures:

    hessian