2 ms·Torax: A Fast and Differentiable Tokamak Transport Simulator in Jax4 points by jasondavies 2y ago