Learning Rate Warmup Scheduler

Implement a function that computes a learning rate schedule combining linear warmup with cosine decay.