Zhang, Shuibai

4 publications

ICLR 2026 ParallelBench: Understanding the Trade-Offs of Parallel Decoding in Diffusion LLMs Wonjun Kang, Kevin Galim, Seunghyuk Oh, Minjae Lee, Yuchen Zeng, Shuibai Zhang, Coleman Richard Charles Hooper, Yuezhou Hu, Hyung Il Koo, Nam Ik Cho, Kangwook Lee

ICLR 2026 Planner Aware Path Learning in Diffusion Language Models Training Fred Zhangzhi Peng, Zachary Bezemek, Jarrid Rector-Brooks, Shuibai Zhang, Michael M. Bronstein, Anru Zhang, Joey Bose, Alexander Tong

ICML 2025 VersaPRM: Multi-Domain Process Reward Model via Synthetic Reasoning Data Thomas Zeng, Shuibai Zhang, Shutong Wu, Christian Classen, Daewon Chae, Ethan Ewer, Minjae Lee, Heeju Kim, Wonjun Kang, Jackson Kunde, Ying Fan, Jungtaek Kim, Hyung Il Koo, Kannan Ramchandran, Dimitris Papailiopoulos, Kangwook Lee

ICLR 2024 Supervised Knowledge Makes Large Language Models Better In-Context Learners Linyi Yang, Shuibai Zhang, Zhuohao Yu, Guangsheng Bao, Yidong Wang, Jindong Wang, Ruochen Xu, Wei Ye, Xing Xie, Weizhu Chen, Yue Zhang