need to learn and play with world models and see what interesting learnings and things we can make out of it https://diamond-wm.github.io/