Related papers: JAX-in-Cell: A Differentiable Particle-in-Cell Cod…
In our effort to facilitate machine learning-assisted computational fluid dynamics (CFD), we introduce the second iteration of JAX-Fluids. JAX-Fluids is a Python-based fully-differentiable CFD solver designed for compressible single- and…
Context. Inferring spectral parameters from X-ray data is one of the cornerstones of high-energy astrophysics, and is achieved using software stacks that have been developed over the last twenty years and more. However, as models get more…
In this work, we present a new scalable incomplete LU factorization framework called Javelin to be used as a preconditioner for solving sparse linear systems with iterative methods. Javelin allows for improved parallel factorization on…
Room-temperature metals and semi-metals which consist of a gas of bound electrons in a near-continuum band structure can be classified as cold quantum plasmas. This insight suggests that Particle-in-Cell (PIC) simulations, traditionally…
This work presents a case study of a heterogeneous multiphysics solver from the nuclear fusion domain. At the macroscopic scale, an auto-differentiable ODE solver in JAX computes the evolution of the pulsed power circuit and bulk plasma…
This paper reports on an in-depth evaluation of the performance portability frameworks Kokkos and RAJA with respect to their suitability for the implementation of complex particle-in-cell (PIC) simulation codes, extending previous studies…
In this article we apply a discrete action principle for the Vlasov--Maxwell equations in a structure-preserving particle-field discretization framework. In this framework the finite-dimensional electromagnetic potentials and fields are…
The 3D quasi-static particle-in-cell (PIC) algorithm is a very efficient method for modeling short-pulse laser or relativistic charged particle beam-plasma interactions. In this algorithm, the plasma response to a non-evolving laser or…
We present a detailed strong and weak scaling analysis of PICCANTE, an open source, massively parallel, fully-relativistic Particle-In-Cell (PIC) code. PIC codes are widely used in plasma physics and astrophysics to study the cases where…
The implicit 2D3V particle-in-cell (PIC) code developed to study the interaction of ultrashort pulse lasers with matter [G. M. Petrov and J. Davis, Computer Phys. Comm. 179, 868 (2008); Phys. Plasmas 18, 073102 (2011)] has been parallelized…
We present TORAX, a new, open-source, differentiable tokamak core transport simulator implemented in Python using the JAX framework. TORAX solves the coupled equations for ion heat transport, electron heat transport, particle transport, and…
We present Brax, an open source library for rigid body simulation with a focus on performance and parallelism on accelerators, written in JAX. We present results on a suite of tasks inspired by the existing reinforcement learning…
In this paper, we present an efficient Particle-In-Cell algorithm for the simulation of the three dimensional Vlasov-Poisson system in the presence of a strong external magnetic field. When the intensity of the magnetic field is…
In order to perform physically faithful particle-in-cell (PIC) simulations, the Gauss's law stands as a critical requirement, since its violation often leads to catastrophic errors in long-term plasma simulations. This work proposes a novel…
A new method that solves concurrently the multi-fluid and Maxwell's equations has been developed for plasma simulations. By calculating the stress tensor in the multi-fluid momentum equation by means of computational particles moving in a…
A recent proof-of-principle study proposes an energy- and charge-conserving, nonlinearly implicit electrostatic particle-in-cell (PIC) algorithm in one dimension [Chen et al, J. Comput. Phys., 230 (2011) 7018]. The algorithm in the…
Recent developments in high peak-power table-top laser systems reaching highly relativistic light intensities have led to significant advances in laser-driven particle acceleration schemes (mainly the laser wakefield acceleration, LWFA)…
Particle methods are a ubiquitous tool for solving the Vlasov-Poisson equation in comoving coordinates, which is used to model the gravitational evolution of dark matter in an expanding universe. However, these methods are known to produce…
A novel adaptive technique for electromagnetic Particle In Cell (PIC) plasma simulations is presented here. Two main issues are identified in designing adaptive techniques for PIC simulation: first, the choice of the size of the particle…
The simulation of problems in kinetic plasma physics are often challenging due to strongly coupled phenomena across multiple scales. In this work, we propose a wavelet-based coarse-grained numerical scheme, based on the framework of…