dq.gradient.Direct

Direct()

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 methods, this falls back to the diffrax.DirectAdjoint option.