braincell.quad.diffrax_heun_step

braincell.quad.diffrax_heun_step#

braincell.quad.diffrax_heun_step(target, *args)[source]#

Advance one step with diffrax’s Heun (improved Euler) solver.

Wraps diffrax.Heun, a two-stage second-order explicit Runge-Kutta method.

Parameters:

target (DiffEqModule) – Module whose differential states will be advanced.
*args – Extra positional arguments forwarded to target’s integration hooks.

Returns:

target’s state is updated in place.

Return type:

None

Raises:

ModuleNotFoundError – If diffrax is not installed.