Size Zheng

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

BMAttn: Block-Aligned Mixed-Precision Attention Quantization for LLM Inference

DITRON: A Flexible and Versatile Distributed Tensor Compiler for LLM

ShadowKV: KV Cache in Shadows for High-Throughput Long-Context LLM Inference

ShadowKV: KV Cache in Shadows for High-Throughput Long-Context LLM Inference

MxMoE: Mixed-precision Quantization for MoE with Accuracy and Performance Co-Design