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

vllm-bot merged 1 commit intovllm-project:mainvllm-project/vllm:mainfrom

hyeygit:topk_tpuhyeygit/vllm:topk_tpuCopy head branch name to clipboard