Probabilistic Deep Learning for Expressive and Reliable World Models