Excited to announce the release of TORAX, a tokamak transport simulator from our @GoogleDeepMind Fusion team! #fusionenergy - Open-source: github.com/google-deepmind/t… - Uses JAX: fast, differentiable - Easy coupling of ML-surrogates Hot off the press → arxiv.org/abs/2406.06718