def METHOD_NAME(test_case):
    arg_dict = OrderedDict()
    arg_dict["device"] = ["cuda"]
    if os.getenv("ONEFLOW_TEST_CPU_ONLY"):
        arg_dict["device"] = ["cpu"]
    arg_dict["x_shape"] = [(10,)]
    arg_dict["learning_rate"] = [0.1, 1e-3]
    arg_dict["train_iters"] = [10]
    arg_dict["betas"] = [(0.99, 0.9)]
    arg_dict["weight_decay"] = [0.001, 0.1]
    arg_dict["eps"] = [1e-6]
    arg_dict["do_bias_correction"] = [True, False]
    arg_dict["adam_w_mode"] = [True, False]
    # NOTE(l1aoxingyu): max_norm = -1 means no clip grad
    arg_dict["clip_grad_max_norm"] = [-1, 0.0, 0.5, 1.0]
    arg_dict["clip_grad_norm_type"] = ["inf", "-inf", 0.0, 1.0, 2.0, 3.5]
    arg_dict["reload_state_step"] = [5]
    arg_dict["save_load_by_pickle"] = [False, True]
    arg_dict["contiguous_params"] = [False, True]
    for arg in GenArgList(arg_dict):
        compare_with_numpy_lamb(test_case, *arg)