← Search

Xuegui Zheng

3 accepted papers

2026

CoCoQuant: Breaking the Bandwidth Wall via Co-Optimized Communication and Computation Quantization

ICML 2026poster

The rapid scaling of large language models (LLMs) has made distributed inference indispensable, yet end-to-end latency is increasingly dominated by communication, forming a critical bandwidth wall that fundamentally limits the practical gains of existing quantization techniques. Existing approaches …

Cited by 0SourceScholar
2026

DITRON: Distributed Multi-level Tiling Compiler for Parallel Tensor Programs

ICML 2026poster

The scaling of large language models (LLMs) is currently bottlenecked by the rigidity of distributed programming. While high-performance libraries like CuBLAS and NCCL provide optimized primitives, they lack the flexibility required for rapidly evolving model architectures. Conversely, existing tens…

Cited by 0SourceScholar
2026

Forge: Compiling a Unified Abstraction into Scalable Kernels for Linear Attention

ICLR 2026poster

The quadratic complexity of softmax attention poses a major bottleneck for long-context modeling, motivating a surge of linear attention variants with linear complexity. Unlike softmax attention, which benefits from optimized kernels, linear attention lacks general-purpose, hardware-efficient suppor…

Cited by 0SourceScholar