← Questions

Top-p (nucleus) sampling in NumPy

0
AmazonCodingLast asked Jun 2026

Implement top-p sampling in NumPy: from a vector of logits, keep the smallest set of most likely tokens whose probabilities add up to at least p, renormalize them, and sample one token.

Write and test your solution herePython in your browser, the question's tests level by level, every submission kept for you.Sign in to start coding