Backend
INC-MoE: Overlapping In-Network Collectives and Expert Computation for Expert-Parallel Mixture-of-Experts Inference
arXiv cs.DC0 views
arXiv:2610.36339v1 Announce Type: new Abstract: In expert-parallel mixture-of-experts (MoE) serving, dispatch moves tokens to the accelerators hosting their selected experts, and combine returns expert outputs to each token's owner. Both operations lie on the critical path and can dominate MoE layer latency in multi-node deployments. We present INC-MoE, an expert-parallel MoE layer that offloads dispatch and combine to the network switch and overlaps both collectives with each other and with expert computation. Dispatch uses in-network AllGather through switch-side multicast, while combine uses in-network ReduceScatter through switch-side reduction. Both collectives use message semantics between the end hosts and the switch. Because the in-network collectives load opposite directions of each link, executing them concurrently reduces bandwidth-bound communication time by $(N-2)/(2(N-1))$ relative to concurrent ring-based collectives: 33% for four ranks and 43% for eight. INC-MoE partitions tokens into bands and processes them through a three-stage AllGather-feed-forward network (FFN)-ReduceScatter pipeline inside a single persistent kernel. At steady state, role-specialized thread blocks gather the next band, compute the current band, and reduce the previous band concurrently, overlapping both collectives with each other and with computation using device-side synchronization. INC-MoE integrates as a drop-in replacement for the MoE layer of vLLM. We evaluate it in an emulated four-node scale-out deployment with a dedicated accelerator emulating an in-network computing (INC) switch. On OLMoE prefill with four-way expert parallelism, INC-MoE reduces time to first token (TTFT) by up to 33% and increases throughput by up to 50% relative to vLLM's default collective path, while matching its outputs.
Source:https://arxiv.org/abs/2610.36339PreviousTRACE: Governing AI-Assisted Research with Principles from Computer ScienceNextIOSage: Detecting HPC I/O Bottleneck Patterns to Guide LLM Recommendations
Related posts

One rebuild, not twenty: what caching a live market dashboard taught me
0dev.to AI
Rate-Limit Error Handling for Structured Candidate Scoring and Moderation APIs
0dev.to AI
DevRel Didn't Die, Its Audience Changed
0dev.to AI
Google ships Gemini 4 Argon, previews Carbon for cyber work
0dev.to AI
Giving AI agents hands without handing over the keys: how we think about safety at TwinMCP
0dev.to AI
Building a Multi-Tenant EKS Platform for 12 Teams: RBAC, Network Policies, and Karpenter with Terraform
0dev.to AWS