brainunit.autograd module

First-order Gradients

vector_grad	Unit-aware element-wise gradient of a vector-valued function.
value_and_grad	Physical unit-aware version of jax.value_and_grad.
grad	Physical unit-aware version of jax.grad.

Jacobian Matrix

jacrev	Physical unit-aware reverse-mode Jacobian of fun.
jacfwd	Physical unit-aware forward-mode Jacobian of fun.
jacobian	Alias of jacrev().

Hessian Matrix

hessian

Physical unit-aware Hessian of fun as a dense array.