about
Jasmine: A Simple, Performant and Scalable Jax-Based World Modeling Codebase (arxiv.org)
24 points by PaulHoule 326 days ago | hide | past | pdf | 1 comment on HN

In plain words: Jasmine is an open toolkit for training world models, which predict how an environment changes, and it scales from one machine to hundreds of AI chips with little code change. It reproduced the CoinRun case study about ten times faster than earlier open versions.

Abstract · Jasmine: A Simple, Performant and Scalable JAX-based World Modeling Codebase

While world models are increasingly positioned as a pathway to overcoming data scarcity in domains such as robotics, open training infrastructure for world modeling remains nascent. We introduce Jasmine, a performant JAX-based world modeling codebase that scales from single hosts to hundreds of accelerators with minimal code changes. Jasmine achieves an order-of-magnitude faster reproduction of the CoinRun case study compared to prior open implementations, enabled by performance optimizations across data loading, training and checkpointing. The codebase guarantees fully reproducible training and supports diverse sharding configurations. By pairing Jasmine with curated large-scale datasets, we establish infrastructure for rigorous benchmarking pipelines across model families and architectural ablations.

Mihir Mahajan, Alfred Nguyen, Franz Srambical, Stefan Bauer
arXiv:2510.27002 · cs.LG, cs.AI · submitted Oct 30, 2025
abstract · pdf · html · Blog post: https://pdoom.org/jasmine.html

add comment on HN

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.