FlashSampling: Fast and Memory-Efficient Exact Sampling
DGX agentarXiv:2603.15854v2 Announce Type: replace-cross Abstract: Sampling from a categorical distribution is mathematically simple, but in large-vocabulary decoding, it often triggers extra memory traffic an