Skip to content

dq.gradient.HigherOrder

HigherOrder()

Higher-order automatic differentiation.

Enables support for higher-order automatic differentiation, such as jax.hessian.

Note

For supported explicit Diffrax ODE methods, this is the documented Hessian-compatible mode. Like dq.gradient.Direct, it uses diffrax.DirectAdjoint, which already configures the solver for higher-order differentiation. Use this mode to make the higher-order differentiation requirement explicit and to restrict to the methods on which it is supported.