Learning Modular Simulations for Homogeneous Systems - Gupta, Jayesh; Vemprala, Sai; Kapoor, Ashish (Advances in Neural Information Processing Systems, 2022)