Adding learning rate decays found in Neural Optimizer Search with...

Adding learning rate decays found in Neural Optimizer Search with Reinforcement Learning [Bello et al, ICML2017]
Also adding cosine decay.

PiperOrigin-RevId: 173451903