Loading…
Cosine Learning Rate Schedule with Warmup is a easy coding interview question covering deep learning, math — the kind asked at FAANG (Facebook/Meta, Apple, Amazon, Netflix, Google) and AI-first companies like Anthropic, OpenAI, and Nvidia.
Implement lr_at_step(step) for the learning-rate schedule used to train GPT and Llama: linear warmup from 0 to the base rate over the first warmup_steps, then cosine decay from the base rate down to a minimum floor over the remaining steps.
Solution walkthrough on Pro
Upgrade to Pro to unlock the guided hints, the step-by-step walkthrough, solutions in Python, C++, Java, and Go, and complexity analysis.
What's inside this guide
Fully unlocked, no account needed — solutions, walkthroughs and follow-up answers included.
Or start the Study Plan — a week-by-week order for your loop, and week 1 is entirely free.
A full interview loop is more than one round. These are the others, each with what is on the site for it.