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.

Note

For supported explicit Diffrax ODE methods, this option also supports higher-order automatic differentiation (e.g. jax.hessian). Prefer dq.gradient.HigherOrder to make that requirement explicit.