MATRIX: Multi-Agent Trajectory Generation with Diverse Contexts. - Xu, Zhuo; Zhou, Rui; Yin, Yida; Gao, Huidong; Tomizuka, Masayoshi; Li, Jiachen (2024)