Wrap up the following code in its own function that depends on the model. This simplifies maintaining the timestepping workaround in OceananigansReactantExt

maybe_initialize_state!(model)

function maybe_initialize_state!(model)
    if model.clock.iteration == 0
        update_state!(model)
    end
    return nothing
end

For all three currently supported timesteppers