← back to the normal site

Jingwei Zuo (左京伟)

I am a second year Ph.D. student in the Computer Science department at Rice University, advised by Dr. Yuke Wang. Prior to this, I obtained bachelor’s degrees in both science and engineering from Tsinghua University.

Multi-agent systems drastically increase inference costs: agent steps, tool iterations, and multi-turn collaboration rapidly blow up KV-cache memory and latency. My research focuses on making these systems computationally viable — DuoAttention and DiLoCo on MaxText on the systems side, AgentVerse on the capability side.

Email: jingwei.zuo [at] rice [dot] edu

Research

My main research question is: How to compute more with less resources?

News

• 05/18/2026 — 🎉 Will join Google Cloud as a PhD SWE Intern!

Publications

• DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Heads
  Guangxuan Xiao, Jiaming Tang, Jingwei Zuo, Junxian Guo, Shang Yang, Haotian Tang, Yao Fu, Song Han
  ICLR 2025
  [paper] [code] [demo]
• AgentVerse: Facilitating Multi-Agent Collaboration and Exploring Emergent Behaviors
  Weize Chen*, Yusheng Su*, Jingwei Zuo, Cheng Yang, Chenfei Yuan, Chi-Min Chan, Heyang Yu, Yaxi Lu, Yi-Hsin Hung, Chen Qian, Yujia Qin, Xin Cong, Ruobing Xie, Zhiyuan Liu, Maosong Sun, Jie Zhou
  ICLR 2024
  [paper] [code]

Experiences

• Google (2026.09-Present)
  Student Researcher, Google Cloud AI Research — Managed by Linfeng Song
• Google, Seattle, WA, USA (2026.05-2026.08)
  SWE Intern, JAX Training Experience Team, Core ML — Google Cloud
• Moonshot AI, Beijing, China (2024.12-2025.05)
  AI Infrastructure Internship — LLM Inference Team
• Carnegie Mellon University, Pittsburgh, PA, USA (2024.07-10)
  Remote Research Internship — InfiniAI Lab, advised by Prof. Beidi Chen
• Massachusetts Institute of Technology, Cambridge, MA, USA (2023.10-2024.05)
  Research Internship — Han Lab, advised by Prof. Song Han
• Tsinghua University, Beijing, China (2023.03-09)
  Research Internship — Natural Language Processing Lab (THUNLP), advised by Prof. Zhiyuan Liu

Education

• Rice University (2025.09-Present)
  • Doctoral Study in Computer Science and Information Systems
• Tsinghua University (2021.09-2025.6)
  • B.Eng. in Electrical Engineering
  • B.S. in Fundamental Sciences (Math & Physics)
• Northeastern University (2023.09-12)
  • Exchange student at College of Engineering
  • Selected on Dean's list

Hobbies

• Sports: Tennis.
• Music: Sometimes I play piano. Classical music and pop music are both my favorites.
• Photography: At the crossing between nature and humanity. Portfolio: (Ig) chris.z_pics
• Reading: books about sociology, psychology, and history, also novels.

← back to the normal site