Shaokun Zhang College of Information Science and Technology Pennsylvania State University E-mail: shaokun.zhang@psu.edu Github   Twitter   Linkedin   Google Scholor

• Nemotron-Research-Tool-N1: Exploring Tool-Using Language Models with Reinforced Reasoning
  Shaokun Zhang, Yi Dong, Jieyu Zhang, Jan Kautz, Bryan Catanzaro, Andrew Tao, Qingyun Wu, Zhiding Yu, Guilin Liu
  Arxiv Preprint, May, 2025
  

• Which Agent Causes Task Failures and When? On Automated Failure Attribution of LLM Multi-Agent Systems
  Shaokun Zhang*, Ming Yin*, Jieyu Zhang, Jiale Liu, Zhiguang Han, Jingyang Zhang, Beibin Li, Chi Wang, Huazheng Wang, Yiran Chen, Qingyun Wu
  Forty-two International Conference on Machine Learning (ICML) (Spotlight Presentation, 313/12108=2.6%)
  

• BEST-Route: Adaptive LLM Routing with Test-Time Optimal Compute
  Dujian Ding, Ankur Mallick, Shaokun Zhang, Chi Wang, Daniel Madrigal, Mirian Del Carmen Hipolito Garcia, Menglin Xia, Laks V. S. Lakshmanan, Qingyun Wu, Victor Rühle
  Forty-two International Conference on Machine Learning (ICML)
  

• Advances and Challenges in Foundation Agents: From Brain-Inspired Intelligence to Evolutionary, Collaborative, and Safe Systems
  …, Shaokun Zhang, et al.
  Arxiv Preprint, Apr, 2025
  

• EcoAct: Economic Agent Determines When to Register What Action
  Shaokun Zhang, Jieyu Zhang, Dujian Ding, Mirian Hipolito Garcia, Ankur Mallick, Daniel Madrigal, Menglin Xia, Victor Ruhle, Qingyun Wu, Chi Wang
  ICLR 2025 Workshop on Reasoning and Planning for LLMs (ICLR@ Reasoning and Planning for LLMs)
  

• Adaptive In-conversation Team Building for Language Model Agents
  Linxin Song, Jiale Liu*, Jieyu Zhang, Shaokun Zhang, Ao Luo, Shijian Wang, Qingyun Wu, Chi Wang
  Arxiv Preprint, May, 2024
  

• Offline Training of Language Model Agents with Functions as Learnable Weights
  Shaokun Zhang*, Jieyu Zhang*, Jiale Liu, Linxin Song, Chi Wang, Ranjay Krishna, Qingyun Wu
  Forty-first International Conference on Machine Learning (ICML) (Media Coverage: The Forbes Article)
  

• Refined Coreset Selection: Towards Minimal Coreset Size under Model Performance Constraints
  Xiaobo Xia, Jiale Liu, Shaokun Zhang, Qingyun Wu, Hongxin Wei, Tongliang Liu
  Forty-first International Conference on Machine Learning (ICML) (Spotlight Presentation, 355/9473=3.74% Collected by OpenAI PaperBench)
  

• StateFlow: Enhancing LLM Task-Solving through State-Driven Workflows
  Yiran Wu, Tianwei Yue, Shaokun Zhang, Chi Wang, Qingyun Wu
  Conference on Language Model (COLM)
  

• IDEAL: Influence-Driven Selective Annotations Empower In-Context Learners in Large Language Models
  Shaokun Zhang*, Xiaobo Xia*, Zhaoqing Wang, Lin-Hao Chen, Jiale Liu, Qingyun Wu, Tongliang Liu
  Eleventh International Conference on Learning Representations (ICLR)
  

• AutoGen: Enabling Next-Gen LLM Applications via Multi-Agent Conversation
  Qingyun Wu, Gagan Bansal, Jieyu Zhang, Yiran Wu, Beibin Li, Erkang Zhu, Li Jiang, Xiaoyun Zhang, Shaokun Zhang, Jiale Liu, Ahmed Hassan Awadallah, Ryen W White, Doug Burger, Chi Wang
  ICLR 2024 Workshop on LLMAgents (Best Paper Award); Conference on Language Model (COLM)
  

• Fair Online Influence Maximization
  Xiangqi Wang, Shaokun Zhang, Jose Efraim Aguilar Escamilla, Qingyun Wu, Xiangliang Zhang, Jian Kang, Huazheng Wang
  Transactions on Machine Learning Research (TMLR)
  

• An Empirical Study on Challenging Math Problem Solving with GPT-4
  Yiran Wu, Feiran Jia, Shaokun Zhang, Hangyu Li, Erkang Zhu, Yue Wang, Yin Tat Lee, Richard Peng, Qingyun Wu, Chi Wang
  ICLR 2024 Workshop on LLMAgents (ICLR@LLMAgents)
  

• Targeted Hyperparameter Optimization with Lexicographic Preferences Over Multiple Objectives
  Shaokun Zhang, Feiran Jia, Chi Wang, Qingyun Wu
  Eleventh International Conference on Learning Representations (ICLR) (Oral Presentation, 91/4955=2.39%)
  

• HyperTime: Hyperparameter Optimization for Combating Temporal Distribution Shifts
  Shaokun Zhang, Yiran Wu, Zhonghua Zheng, Qingyun Wu, Chi Wang
  ACM Multimedia (ACM MM)
  

• You Only Compress Once: Towards Effective and Elastic BERT Compression via Exploit-Explore Stochastic Nature Gradient
  Shaokun Zhang, Xiawu Zheng, Chenyi Yang, Yuchao Li, Yan Wang, Fei Chao, Mengdi Wang, Shen Li, Jun Yang, Rongrong Ji
  Neurocomputing
  

• Efficient Neural Architecture Search via Dynamic Distribution Pruning
  Xiawu Zheng, Chenyi Yang, Shaokun Zhang, Yan Wang, Baochang Zhang, Yongjian Wu, Yunsheng Wu, Ling Shao, Rongrong Ji
  International Journal of Computer Vision 131.5 (2023): 1234-1249 (IJCV)
  

(Sorted by appeard date, *equal contribution)

• Autogen: LLM agents
  Role: Creator. (40k stars, Open100: Top 100 Open Source achievements)
  
• FLAML: AutoML and hyperparameter tuning
  Role: Maintainer. (3.3k stars, 1 millon downloads)
  

• Nvidia (Santa Clara), Research Intern
  2025.02 - Present
  Topic: LLMs Post-training.
• Microsoft Research (Redmond), Research Intern
  2024.06 - 2024.09
  Topic: Language Agent.

• Reinforcement Learning Open-Source Fest Award, Microsoft Research at NYC, 2022.
• Chinese National Scholorship, Ministry of Education of China, 2018.

• DS 440: Data Sciences Capstone Course
  Instructor/Teaching Assistant, Penn State, Spring 2023 & Fall 2023.
• IST 866: Threat Hunting and Offensive Cybersecurity
  Instructor/Teaching Assistant, Penn State, Fall 2025.
• ETI 463: Distributed Database Management Systems
  Instructor/Teaching Assistant, Penn State, Fall 2025.

• Conference Reviewer:
• ICLR'24-26, NeurIPS'23-25, ICML'24'25
• ACL'23'25, UAI'23, AutoML'23, KDD'24'25, AAAI'25
• AJCAI'23, ICIC'23
• Journal Reviewer:
• TCSVT
• Information Fusion