Waymax: An Accelerated, Data-Driven Simulator for Large-Scale Autonomous Driving Research - Gulino, Cole; Fu, Justin; Luo, Wenjie; Tucker, George; Bronstein, Eli; Lu, Yiren; Harb, Jean; Pan, Xinlei; Wang, Yan; Chen, Xiangyu; Co-Reyes, John; Agarwal, Rishabh; et al. (Advances in Neural Information Processing Systems, 2023)