profile photo
Yujie Zhao 赵雨洁
CSE Ph.D. Student @ UCSD
yuz285[at]ucsd.edu
zhaoyj04[at]gmail.com

Description of the icon About

Hey there! I'm Yujie Zhao, a second-year Ph.D. student advised by Prof. Jishen Zhao in Computer Science and Engineering Department of UC San Diego, starting in 2024. I am now aiming to build efficient systems for machine learning. Before coming to UCSD, I received my B.S. degree in Applied Physics at University of Science and Technology of China. I used to be an undergraduate research intern in EIC Lab at GaTech under the supervision of Professor Yingyan (Celine) Lin and Yang (Katie) Zhao. Previsouly I have also done research about RL theory under the supervision of Professor Huazheng Wang. I have a broad interest in combining machine learning and computer systems, with a special focus on hardware-algorithm co-design. Feel free to contact me if you share any common interests!

Description of the icon Recent Research Interests

1. Multi-Agent Systems, including inference and training.
2. RL training for LLM agents, including efficiency and effectiveness.

Description of the icon News

  • Excited to release our PettingLLMs for multi-agent reinforcement learning training framework — check it out and give it a try! 🚀
  • Thrilled to share our new preprint on on-policy reinforcement learning for multi-agent systems! Check out the blog for more details. 📄
  • Completed three-month research internship at Intel AI Team.
  • One first-author paper accepted to DAC 2025.
  • One first-author paper accepted to NeurIPS 2024.
  • One first-author paper accepted to DAC 2024.

Description of the icon First Author Publications (Full List on Google Scholar)

Stronger Together: On-Policy Reinforcement Learning with Multi-Agent Systems

Yujie Zhao, Lanxiang Hu, Yang Wang, Minmin Hou, Hao Zhang, Ke Ding, Jishen Zhao.

RA-PbRL: Provably Efficient Risk-Aware Preference-Based Reinforcement Learning

Yujie Zhao, Jose Aguilar Escamilla, Weyl Lu, Huazheng Wang.
Accepted by NeurIPS 2024, CCF A

MAGE: A Multi-Agent Engine for Automated RTL Code Generation

Yujie Zhao* , Hejia Zhang* , Hanxian Huang, Zhongming Yu, Jishen Zhao
Accepted by DAC 2025, CCF A
3D-Carbon: An Analytical Carbon Modeling Tool for 3D and 2.5D Integrated Circuits

Yujie Zhao, Yang Zhao, Cheng Wan, Yingyan (Celine) Lin
Accepted by DAC 2024, CCF A

Description of the icon Selected Collaborated Publications and Manusripts

1. OrcaLoca: An LLM Agent Framework for Software Issue Localization
Zhongming Yu*, Hejia Zhang*,Yujie Zhao, Hanxian Huang, Matrix Yao, Ke Ding, Jishen Zhao
📄 Paper💻 Code
2. Instant-RAG: Enabling Instant Retrieval-Augmented Generation in LLMs via In-Storage Processing
Zhongzhi Yu, Yujie Zhao, Sixu Li, Yujie Wang, Chaojian Li, Amrit Khera, Shruti Belgali, Yang Zhao, Yingyan (Celine) Lin
Top