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