← Search

Ningxin Zheng

5 accepted papers

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
2025

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

ICML 2025spotlight

With the widespread deployment of long-context large language models (LLMs), there has been a growing demand for efficient support of high-throughput inference. However, as the key-value (KV) cache expands with the sequence length, the increasing memory footprint and the need to access it for decodi…

2023

EfficientViT: Memory Efficient Vision Transformer With Cascaded Group Attention

CVPR 2023poster

Vision transformers have shown great success due to their high model capabilities. However, their remarkable performance is accompanied by heavy computation costs, which makes them unsuitable for real-time applications. In this paper, we propose a family of high-speed vision transformers named Effic…

2023

SpaceEvo: Hardware-Friendly Search Space Design for Efficient INT8 Inference

ICCV 2023poster

The combination of Neural Architecture Search (NAS) and quantization has proven successful in automatically designing low-FLOPs INT8 quantized neural networks (QNN). However, directly applying NAS to design accurate QNN models that achieve low latency on real-world devices leads to inferior performa…

Cited by 5PDFcodeScholar