GEM: Group Enhanced Model for Learning Dynamical Control Systems. - Hansen-Estruch, Philippe; Shang, Wenling; Pinto, Lerrel; Abbeel, Pieter; Tiomkin, Stas (CoRR, 2021)