1 citations · 1 across the 2 of their papers we have counts for
2 papers
cs.DC2026★ 1 cited
MPK: A Compiler and Runtime for Mega-Kernelizing Tensor Programs
Xinhao Cheng, Zhihao Zhang, Yu Zhou +17
We introduce Mirage Persistent Kernel (MPK), the first compiler and runtime system that automatically transforms multi-GPU model inference into a single high-performance mega-kerne…
cs.LG2026
Uncertainty in a Single Pass: A Closed-Form Identity for One-Step Flow Matching
Jiarui Xing, Song Wang, Jian Wang
Flow matching provides a highly effective framework for generative modeling, yet estimating the uncertainty of its generated samples remains a fundamental challenge. Existing metho…