flashinfer-ai / flashinfer

FlashInfer: Kernel Library for LLM Serving
https://flashinfer.ai
Apache License 2.0
822 stars 77 forks source link

misc: parallel sampling from probability #214

Closed yzh119 closed 2 months ago

yzh119 commented 2 months ago

follow up of #213 , add the parallel version of parallel sampling without top-p restriction.