TORAX is a differentiable tokamak core transport simulatorgithub.com/google-deepmind105 pointsyeldarb2 years ago