Bansor: Improving Tensor Program Auto-Scheduling with Bandit Based Reinforcement Learning. - Chao Gao; Tong Mo; Taylor Zowtuk; Tanvir Sajed; Laiyuan Gong; Hanxuan Chen; Shangling Jui; Wei Lu 023 (ICTAI, 2021)

A Memory-Bounded Best-First Beam Search and Its Application to Scheduling Halide Programs. - Chao Gao; Jingwei Chen; Tong Mo; Tanvir Sajed; Shangling Jui; Min Qin; Laiyuan Gong; Wei Lu 023 (SOCS, 2022)

CASCO: Cascaded Co-Optimization for Holistic Neural Network Acceleration. - Bahador Rashidi; Shan Lu; Kiarash Aghakasiri; Chao Gao; Fred Xuefei Han; Zhisheng Wang; Laiyuan Gong; Fengyu Sun (DATE, 2024)