Wentao Ni 🎵
Wentao Ni

First-year Ph.D.

About Me

Wentao Ni is a first year PhD at STABLE lab in University of California San Diego, advised by Prof. Jishen Zhao. His research interests include machine learning system, high performance computing and computer architecture. He wants to contribute to faster computations and less resources needed in machine learning and other computation tasks.

Download CV
Interests
  • Machine Learning System
  • High Performance Computing
  • Computer Architecture
Education
  • Ph.D.

    University of California San Diego, CSE

  • B.E. in Computer Science

    Shanghai Jiao Tong University, ACM Honors Class

📚 My Research

Currently I am interested in long context inference and RAG systems, with respect to approximate nearest neighbor search and serving systems.

Please reach out to collaborate 😃

Recent Publications
(2025). VQ-LLM: High-performance Code Generation for Vector Quantization Augmented LLM Inference. HPCA 2025.
(2024). JUNO: Optimizing High-Dimensional Approximate Nearest Neighbour Search with Sparsity-Aware Algorithm and Ray-Tracing Core Mapping. ASPLOS 2024.