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