DeepOffer

Implement top-k, top-p, and temperature sampling over a logits vector.

ML CodingReported interview question
Reported in public interview compilations — Google DeepMind, Apple

Scale logits by temperature, mask outside top-k or the cumulative top-p set, subtract the maximum for stability, softmax, and sample. Handle zero temperature deterministically and renormalize after filtering.

Test empty input, one-element input, duplicates, boundary indices, invalid states, and the largest allowed size; state time and space complexity.

Common follow-up questions

Practice this question with an AI interviewer

Get asked follow-ups live, then receive a scored report — like a real MLE interview loop.

Start AI mock interview