flashinfer-ai / flashinfer

FlashInfer: Kernel Library for LLM Serving
https://flashinfer.ai
Apache License 2.0
1.22k stars 115 forks source link

feat: initial support of logits hook #298

Closed yzh119 closed 3 months ago

yzh119 commented 3 months ago

Implement the #257 feature.