braincell.quad.diffrax_ralston_step

braincell.quad.diffrax_ralston_step#

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

Advance one step with diffrax’s Ralston second-order solver.

Wraps diffrax.Ralston, the second-order explicit Runge-Kutta method that minimises the leading-order truncation error coefficient.

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

ralston2_step

Native (no-diffrax) Ralston implementation.