Ian Goodfellow

TORAX: A Fast and Differentiable Tokamak Transport Simulator in JAX

TORAX：JAX 中一个快速且可微的托卡马克输运模拟器

Jonathan Citrin, Ian Goodfellow, Akhil Raju, Jeremy Chen, Jonas Degrave, Craig Donner, Federico Felici, Philippe Hamel, Andrea Huber, Dmitry Nikulin, David Pfau, Brendan Tracey, Martin Riedmiller, Pushmeet Kohli

AI总结 介绍基于 JAX 框架的开源可微托卡马克输运模拟器 TORAX，它求解多种输运耦合方程，融合物理与机器学习模型，利用 JAX 特性实现快速运行及微分，经与 RAPTOR 代码验证，为托卡马克相关研究提供有力工具。

Comments 16 pages, 7 figures