Swirl-Jatmos: The Atmospheric Simulator That Lets JAX Do the MPI
A TPU-first Python framework for 3D weather physics, where sharding, pressure solves, and checkpoints are handled as compiler-aware infrastructure, not hand-built cluster code.
10 min read · Apr 7, 2026