Workshop on Machine Learning and Automatic Differentiation in JAX for Scientific Computing
Strasbourg, France
June 8 - 12, 2026
A JAX journey in building TORAX for plasma physics simulation
Sebastian
Bodenstein & Anushan
Fernando (Google DeepMind, London, UK)