Hyesoo Yang 47195057e9 [V1][TPU] Speed up top-k on TPU by using torch.topk (#15242) ...

Signed-off-by: Hyesoo Yang <hyeygit@gmail.com>

2025-03-20 19:19:40 -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

[V1] Support bad_words in sampler (#13376)

2025-03-08 14:50:26 -08:00

penalties.py

Update deprecated Python 3.8 typing (#13971)

2025-03-02 17:34:51 -08:00

topk_topp_sampler.py

[V1][TPU] Speed up top-k on TPU by using torch.topk (#15242)

2025-03-20 19:19:40 -07:00

utils.py

[V1][Spec Decode] Optimize Rejection Sampler with Triton Kernels (#14930)

2025-03-18 14:31:54 -07:00