dq.gradient.Autograd

Autograd()

Standard automatic differentiation of JAX.

With this option, the gradient is computed by automatically differentiating through the internals of the solver.

Note

For Diffrax-based solvers, this falls back to the diffrax.DirectAdjoint option.