def METHOD_NAME(model, base_scale):
    param_groups = []
    for i, layer in enumerate(model.layers):
        this_scale = base_scale ** (i + 1)
        param_group = {"params": layer.parameters(), "lr_scale": this_scale}
        param_groups.append(param_group)
    return param_groups