8 citations · 10 across the 7 of their papers we have counts for
Showing 2026Show all
3 papers · 1 filter
cs.LG2026
PithTrain: A Compact and Agent-Native MoE Training System
Ruihang Lai, Hao Kang, Haozhan Tang +6
Mixture-of-Experts (MoE) has become the dominant architecture for frontier language models. To meet this demand, production frameworks have built optimized MoE training stacks over…
cs.DC2026
Axe: A Simple Unified Layout Abstraction for Machine Learning Compilers
Bohan Hou, Hongyi Jin, Guanjie Wang +7
Scaling modern deep learning workloads demands coordinated placement of data and compute across device meshes, memory hierarchies, and heterogeneous accelerators. We present Axe La…
cs.AI2026★ 1 cited
XGrammar-2: Dynamic and Efficient Structured Generation Engine for Agentic LLMs
Linzhang Li, Yixin Dong, Guanjie Wang +3
Modern LLM agents increasingly rely on dynamic structured generation, such as tool calling and response protocols. Unlike traditional structured generation with static structures,…