Pu, Yuan

6 publications

ICLR 2026 One Model for All Tasks: Leveraging Efficient World Models in Multi-Task Planning Yuan Pu, Yazhe Niu, Jia Tang, Junyu Xiong, Shuai Hu, Hongsheng Li

ICLR 2025 Circuit Representation Learning with Masked Gate Modeling and Verilog-AIG Alignment Haoyuan Wu, Haisheng Zheng, Yuan Pu, Bei Yu

TMLR 2025 UniZero: Generalized and Efficient Planning with Scalable Latent World Models Yuan Pu, Yazhe Niu, Zhenjie Yang, Jiyuan Ren, Hongsheng Li, Yu Liu

NeurIPS 2024 Trajectory Flow Matching with Applications to Clinical Time Series Modelling Xi Zhang, Yuan Pu, Yuki Kawamura, Andrew Loza, Yoshua Bengio, Dennis L. Shung, Alexander Tong

IJCAI 2023 FastGR: Global Routing on CPU-GPU with Heterogeneous Task Graph Scheduler (Extended Abstract) Siting Liu, Yuan Pu, Peiyu Liao, Hongzhong Wu, Rui Zhang, Zhitang Chen, Wenlong Lv, Yibo Lin, Bei Yu

NeurIPS 2023 LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios Yazhe Niu, Yuan Pu, Zhenjie Yang, Xueyan Li, Tong Zhou, Jiyuan Ren, Shuai Hu, Hongsheng Li, Yu Liu