def heun_rk2(equation: ExplicitNavierStokesODE, time_step: float) -> TimeStepFn:
    return jax.named_call(navier_stokes_rk(ButcherTableau(a=[[1]], b=[(1 / 2), (1 / 2)]), equation=equation, time_step=time_step), name='heun_rk2')
def heun_rk2(equation: ExplicitNavierStokesODE, time_step: float) -> TimeStepFn:
    return jax.named_call(navier_stokes_rk(ButcherTableau(a=[[1]], b=[(1 / 2), (1 / 2)]), equation=equation, time_step=time_step), name='heun_rk2')