google-deepmind/torax

Here's a brief summary of the project and its value proposition: **TORAX: Tokamak Transport Simulation in JAX** TORAX is an open-source project that simulates T