collaborators

5 papers

cs.AR2026

MemExplorer: Navigating the Heterogeneous Memory Design Space for Agentic Inference NPUs

Haoran Wu, Zeyu Cao, Yao Lai +15

Emerging agentic LLM workloads are driving rapidly growing demand on both memory capacity and bandwidth, with different phases of inference (e.g., prefill and decode) imposing dist…

cs.AI2026

Heterogeneous Computing: The Key to Powering the Future of AI Agent Inference

Yiren Zhao, Junyi Liu

AI agent inference is driving an inference heavy datacenter future and exposes bottlenecks beyond compute - especially memory capacity, memory bandwidth and high-speed interconnect…

cs.AR2026

NPU Design for Diffusion Language Model Inference

Binglei Lou, Haoran Wu, Kevin Lau +9

Diffusion-based LLMs (dLLMs) fundamentally depart from traditional autoregressive (AR) LLM inference: they leverage bidirectional attention, block-wise KV cache refreshing, cross-s…

cs.AR2025

Combating the Memory Walls: Optimization Pathways for Long-Context Agentic LLM Inference

Haoran Wu, Can Xiao, Jiayi Nie +15

LLMs now form the backbone of AI agents across a diverse range of applications, including tool use, command-line interfaces, and web or computer interaction. These agentic LLM infe…

cs.AR2025

AMPLE: Event-Driven Accelerator for Mixed-Precision Inference of Graph Neural Networks

Pedro Gimenes, Yiren Zhao, George Constantinides

Graph Neural Networks (GNNs) have recently gained attention due to their performance on non-Euclidean data. The use of custom hardware architectures proves particularly beneficial…