braincell.quad.diffrax_midpoint_step

braincell.quad.diffrax_midpoint_step(target, *args)

Advance one step with diffrax’s explicit midpoint solver.

Wraps diffrax.Midpoint, 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.

See also

midpoint_step

Native (no-diffrax) midpoint implementation.