Hyesoo Yangandroot 1b84eff03a [V1][TPU] TPU-optimized top-p implementation (avoids scattering). (#15736) ...

Signed-off-by: Hyesoo Yang <hyeygit@gmail.com>
Co-authored-by: root <root@t1v-n-822696b7-w-0.us-central2-b.c.tpu-prod-env-large-adhoc.internal>

2025-04-02 17:18:08 -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] TPU-optimized top-p implementation (avoids scattering). (#15736)

2025-04-02 17:18:08 -07:00