3 people 1484266938 [Perf] Improve _topk_topp_kernel() for masked logits (#61213) ...

Signed-off-by: Tomas Ruiz <tomas.ruiz.te@gmail.com>
Co-authored-by: Codex <noreply@openai.com>
Co-authored-by: Nick Hill <nickhill123@gmail.com>

2026-10-12 16:01:11 -07:00

..

__init__.py

[V1] Use FlashInfer Sampling Kernel for Top-P & Top-K Sampling (#11394)

2024-12-27 09:32:38 +09:00

bad_words.py

[Perf][1/n] Eliminate various GPU<->CPU syncs (#41429)

2026-05-11 20:36:03 +00:00

logprobs.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

penalties.py

Add pydocstyle to the ruff rules (#52136)

2026-09-16 16:44:29 +00:00

topk_topp_sampler.py

[Bugfix][CPU] Fix fused Gumbel-max wrapping-window bias (#60221)

2026-10-08 15:15:53 +08:00

topk_topp_triton.py

[Perf] Improve _topk_topp_kernel() for masked logits (#61213)

2026-10-12 16:01:11 -07:00