awesome-ml-in-plasma-physics

Automatic-Differentiation-Enabled Plasma Transport code in JAX, enabling differentiable simulation and neural network training pipelines for kinetic and fluid plasma models (

Integrated inter-disciplinary design tool for future fusion reactors with modules for plasma physics, engineering, and optimization (LGPL-2.1)

Python/Rust quasi-steady electromagnetics toolkit covering filament models, Biot-Savart calculations, and Grad-Shafranov utilities

Commonwealth Fusion Systems public repositories for SPARC physics inputs, POPCON analysis, electromagnetics, and scientific-software utilities

Plasma Operating CONtour analysis tool for tokamak performance prediction and optimization

Proxima Fusion dataset of QI-like stellarator boundaries, ideal-MHD equilibria, and optimization benchmarks (

DOI - We present the design and application of a general algorithm for Prediction And Control using MAchiNe learning (PACMAN) in DIII-D. Machine learing (ML)-based predictors and controllers have shown great promise in achieving regimes in which traditional controllers fail, such as tearing mode free scenarios, ELM-free scenarios and stable advanced tokamak conditions. The architecture presented here was deployed on DIII-D to facilitate the end-to-end implementation of advanced control experiments,...

DOI - Accurately predicting plasma behavior based on discharge configurations is essential for the safe and efficient operation of tokamak experiments. While physics-based integrated modeling codes provide valuable insights, their high computational cost limits their applicability for fast scenario design and control optimization. In this study, we propose a transformer-based machine learning model to predict key global plasma parameters on the Tungsten (W) Environment in Steady-State Tokamak (WEST...

DOI - Magnetic equilibrium reconstruction provides the plasma state estimate required for real-time shape control in tokamaks. We present a fast, physics-informed neural network surrogate of the LIUQE equilibrium reconstruction code for the TCV tokamak at EPFL, achieving inference times below 100 microseconds and enabling 10 kHz shape control. The model is trained on around 10,000 TCV discharges spanning the full operational range of plasma shapes. Its modular branch/trunk architecture decouples ma...

DOI - Magnetic geometry has a significant effect on the level of turbulent transport in fusion plasmas. Here, we model and analyse this dependence using multiple machine learning methods and a dataset of ${\gt}200,000$ nonlinear gyrokinetic simulations of ion-temperature-gradient turbulence in diverse non-axisymmetric geometries. The dataset is generated using a large collection of both optimised and randomly generated stellarator equilibria. At fixed gradients and other input parameters, the turb...

DOI - We have developed TorbeamNN: a machine learning surrogate model for the TORBEAM ray tracing code to predict electron cyclotron heating (ECH) and current drive locations in tokamak plasmas. TorbeamNN provides more than a 100 times speed-up compared to the highly optimized and simplified real-time implementation of TORBEAM without any reduction in accuracy compared to the offline, full fidelity TORBEAM code. The model was trained using KSTAR ECH mirror geometries and works for both O-mode and X...

DOIAIP - A machine learning approach has been implemented to measure the electron temperature directly from the emission spectra of a tokamak plasma. This approach utilized a neural network (NN) trained on a dataset of 1865 time slices from operation of the DIII-D tokamak using extreme ultraviolet / vacuum ultraviolet (EUV/VUV) emission spectroscopy matched with high-accuracy divertor Thomson scattering measurements of the electron temperature, $T_e$. This NN is shown to be particularly good at predic...

First open-source foundation model for tokamak plasma dynamics, pretrained on MAST heterogeneous diagnostics

Neural network model for cross-device line-integral diagnostics with physics constraints