Learning Robust Dynamics through Variational Sparse Gating - Jain, Arnav Kumar; Sujit, Shivakanth; Joshi, Shruti; Michalski, Vincent; Hafner, Danijar; Ebrahimi Kahou, Samira (Advances in Neural Information Processing Systems, 2022)