def METHOD_NAME(
    params,
    grads,
    momentums,
    learning_rate=0.0001,
    beta_1=0.9,
    beta_2=0.99,
):
    params = params - learning_rate * np.sign(
        beta_1 * momentums + (1 - beta_1) * grads
    )
    momentums = beta_2 * momentums + (1 - beta_2) * grads
    return params, momentums