Schedulers# step_decay(init, decay_rate, step_size) Make a step decay scheduler. exponential_decay(init, decay_rate) Make an exponential decay scheduler. cosine_decay(init, decay_steps) Make a cosine decay scheduler.