Skip to content
KernelIndex
Search⌘K

Top k sampling from probs v202048

0 eligible runs
sampling

Top-k sampling from probabilities with vocab_size=202048. Keeps only the k highest probability tokens, renormalizes, then samples from the filtered distribution. Captured from Llama 4 Scout/Maverick.

Current records

No published measurement for the selected workload.

Implementations

Implementation
Runtime
Best latency
Evidence
Availability
python
—
—
Apache-2.0 · source

Semantics

Inputs and outputs
probsfp32 [batch_size, vocab_size]
top_kint32 [batch_size]
samplesint64 [batch_size]
Axes and behavior
batch_sizevariable
vocab_sizeconstant = 202048
determinismunspecified
constraintsNo mutation or aliasing
Identity
aliastop_k_sampling_from_probs_v202048modelllama-4-scoutsha256e48126a112ed…
No source imports for this operation yet.How records are decidedJSON