Building efficient AI inference systems
through hardware–software co-design
Chiyue Wei is a Ph.D. candidate in Electrical and Computer Engineering at Duke University, advised by Professor Yiran Chen. His research lies at the intersection of computer architecture and deep learning, focusing on efficient inference systems and hardware–software co-design for artificial intelligence. Prior to Duke, he earned his B.Eng. in Electronic Engineering from Tsinghua University in 2023, where he worked with Professor Yuan Xie and Professor Yu Wang.
Latest
Started my research internship in the AI-System Co-Design team at Meta, where I work on MTIA, Meta’s in-house AI accelerator.
Check out our work DPad, a training-free acceleration method for Diffusion LLMs, now available on arXiv!
Wrapped up my internship at NVIDIA, where I worked on the FlashInfer project — developing high-performance, customizable attention kernels with CuTe DSL, optimized for Blackwell GPUs.
Honored to be named a DAC 2025 Young Fellow.
Our works Phi, Transitive Array, and Ecco were presented at ISCA 2025 — check out the slides for Phi.
Started my summer internship at NVIDIA, focusing on LLM inference framework optimization within the Deep Learning Frameworks team.
Three papers accepted by ISCA 2025! Topics include acceleration for Spiking Neural Networks, General Matrix Multiplications, and Large Language Models.
Our paper Prosperity: Accelerating Spiking Neural Networks via Product Sparsity is accepted by HPCA 2025.
Research
A selection of my work on efficient architectures and systems for artificial intelligence.
Journey
Research Scientist Intern, AI-System Co-Design
Meta · Menlo Park, CA
Architectural design for the next-gen Meta Training and Inference Accelerator (MTIA) targeting LLM inference on a wafer-scale system. Mentors: Aravind Sukumaran-Rajam, Yitu Wang; Manager: Harsha Jagannati.
Deep Learning Intern
NVIDIA · Santa Clara, CA
Built high-performance, customizable attention kernels for Blackwell GPUs using CUTLASS / CuTe DSL, improving inference efficiency in open-source serving stacks (vLLM, FlashInfer). Mentors: Zihao Ye, Pavani Majety; Managers: Cliff Woolley, Kushan Ahmadian.
Graduate Research Assistant
Duke University · Durham, NC
Efficient inference architectures and systems for artificial intelligence. Advisor: Prof. Yiran Chen.
Research Assistant
UC Santa Barbara · Santa Barbara, CA
Systolic-array-friendly design for sparse LU factorization. Advisor: Prof. Yuan Xie.
Research Assistant
Tsinghua University · Beijing, China
Near-memory-computing architectures for graph mining and GNN-based subgraph counting. Advisor: Prof. Yu Wang.
Beyond research