Heteroscedastic Uncertainty for Robust Generative Latent Dynamics - View it on GitHub