Makes me wonder when an update to the world models paper comes out where they drop in diffusion models: https://worldmodels.github.io/