A LearningRateSchedule that uses a cosine decay with optional warmup.
LearningRateSchedule
Useful links