braincell.quad.diffrax_kvaerno3_step

braincell.quad.diffrax_kvaerno3_step#

braincell.quad.diffrax_kvaerno3_step(target, *args, tol=1e-05)[source]#

Advance one step with diffrax’s Kvaerno 3 ESDIRK solver.

Wraps diffrax.Kvaerno3, a third-order, \(L\)-stable explicit-singly-diagonally-implicit Runge-Kutta (ESDIRK) method designed for stiff systems. Implicit stages are solved by a diffrax.VeryChord chord iteration.

Parameters:
  • target (DiffEqModule) – Module whose differential states will be advanced.

  • *args – Extra positional arguments forwarded to target’s integration hooks.

  • tol (float, optional) – Combined relative and absolute tolerance for the chord root-finder. Defaults to 1e-5.

Returns:

target’s state is updated in place.

Return type:

None

Raises:

ModuleNotFoundError – If diffrax is not installed.