[Feature]: Any plans to support TokenWeave optimizations in vLLM?

[solitude-s]

🚀 The feature, motivation and pitch

I recently came across an issue where someone mentioned that TokenWeave achieves performance gains over vLLM by overlapping compute and communication, and by using fused kernels (AllReduce, Residual addition, and RMSNorm).
Are there any plans to support these optimizations in vLLM itself?

vLLM issue: #19415
TokenWeave Code: https://github.com/microsoft/tokenweave/tree/main
TokenWeave Paper: https://arxiv.org/abs/2505.11329

Alternatives

No response

Additional context

No response

Before submitting a new issue...

• Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.

[rajagond]

We are in the process of integrating this.

[houseroad]

Wondering if there is progress on this? :-)

[rajagond]

Hi, I had a discussion with @ProExpertProg, @LucasWilkinson, and others. They said they would be happy to work on integrating this, but I am not sure about the current progress.

[github-actions]

This issue has been automatically marked as stale because it has not had any activity within 90 days. It will be automatically closed if no further activity occurs within 30 days. Leave a comment if you feel this issue should remain open. Thank you!

[ProExpertProg]

Also see nanoflow which has a similar concept of compute-communication overlap: #23592