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 - 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 - This paper presents the development and experimental validation of a reinforcement learning (RL)-based magnetic controller on the DIII-D tokamak. The controller directly maps raw magnetic diagnostic signals to actuator commands, replacing the traditional isoflux control algorithm based on equilibrium reconstruction. Four RL controllers are trained using the Soft Actor–Critic algorithm with an asymmetric Actor–Critic architecture in the NSFsim simulator. All controllers are deployed in the DII...

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 - 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,...

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...

DOILink - We present a fast and accurate data-driven surrogate model for divertor plasma detachment prediction leveraging the latent feature space concept in machine learning research. Our approach involves constructing and training two neural networks. An autoencoder that finds a proper latent space representation (LSR) of plasma state by compressing the multi-modal diagnostic measurements, and a forward model using multi-layer perception (MLP) that projects a set of plasma control parameters to its c...

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