I’m a first year PhD student @GeorgiaTech, co-advised by Bo Dai and Chao Zhang. Before this, I graduated with a Bachelor’s degree from Tsinghua University.

My research interest includes LLM, Modular Learning and AI Agents.

🔥 News

• May 2025 — MLE-Dojo Released!
  A new gym for benchmarking and training AI agents on real-world MLE tasks!

• May 2025 — InF-Embed Released!
  A large-scale, high-quality training corpus and a training recipe tailored for enhancing retrieval models in Instruction-Following IR!

📝 Publications and Preprints

• MLE-Dojo: Interactive Environments for Empowering LLM Agents in Machine Learning Engineering, Rushi Qiang^*, Yuchen Zhuang^*, Yinghao Li, Dingu Sagar V K, Rongzhi Zhang, Changhao Li, Ian Shu-Hei Wong, Sherry Yang, Percy Liang, Chao Zhang, Bo Dai. Preprints.
• Towards Better Instruction Following Retrieval Models, Yuchen Zhuang^*, Aaron Trinh^*, Rushi Qiang^*, Haotian Sun, Chao Zhang, Hanjun Dai, Bo Dai. Preprints.
• AutoLoRA: Automatically Tuning Matrix Ranks in Low-Rank Adaptation Based on Meta Learning, Ruiyi Zhang^*, Rushi Qiang^*, Sai Ashish Somayajula, Pengtao Xie. NAACL 2024
• HYDRA: Model Factorization Framework for Black-Box LLM Personalization, Yuchen Zhuang, Haotian Sun, Yue Yu, Rushi Qiang, Qifan Wang, Chao Zhang, Bo Dai. NeurIPS 2024
• Matryoshka: Learning to Drive Black-Box LLMs with LLMs, Changhao Li^*, Yuchen Zhuang^*, Rushi Qiang, Haotian Sun, Hanjun Dai, Chao Zhang, Bo Dai. Preprints.
• Language Model Uncertainty Quantification with Attention Chain, Yinghao Li, Rushi Qiang, Lama Moukheiber, Chao Zhang. Preprints.

📖 Educations

• 2024.08 - present, Georgia Tech Machine Learning PhD student.
• 2020.09 - 2024.06, B.Eng. in Automation, Tsinghua University..