1 code implementation • 12 Apr 2024 • Jonas Niederle, Simon Koop, Marc Pagès-Gallego, Vlado Menkovski
We introduce an auxiliary regressor on the latent variable to encourage our model to learn an informative latent representation.
1 code implementation • NeurIPS 2023 • Koen Minartz, Yoeri Poels, Simon Koop, Vlado Menkovski
However, to incorporate symmetries in probabilistic neural simulators that can simulate stochastic phenomena, we need a model that produces equivariant distributions over trajectories, rather than equivariant function approximations.