Add Lion baseline

[tfaod]

• Implementation: Google AutoML's Lion
• Tunable hyperparameters: learning_rate, weight_decay, one_minus_beta1, beta2
• Fixed hyperparameters: warmup_factor

[tfaod]

I'm unable to assign this to myself.

cc: @priyakasimbeg

[tfaod]

EDIT - Able to able to myself. Thanks @priyakasimbeg