Jasmine: A Simple, Performant and Scalable Jax-Based World Modeling Codebase

This gets way outside my wheelhouse quick so there is likely a lot that I'm not aware of in this space but this alone "The codebase guarantees fully reproducible training and supports diverse sharding configurations" sounds really compelling from a training perspective.

It really seems like from a consumer perspective robotics is starting to take much bigger leaps forward and I can't begin to imagine what happens when these world models really take off.