English
Related papers

Related papers: JAX-in-Cell: A Differentiable Particle-in-Cell Cod…

200 papers

This paper introduces JAX-FEM, an open-source differentiable finite element method (FEM) library. Constructed on top of Google JAX, a rising machine learning library focusing on high-performance numerical computing, JAX-FEM is implemented…

Mathematical Software · Computer Science 2023-06-28 Tianju Xue , Shuheng Liao , Zhengtao Gan , Chanwook Park , Xiaoyu Xie , Wing Kam Liu , Jian Cao

Implicit particle-in-cell codes offer advantages over their explicit counterparts in that they suffer weaker stability constraints on the need to resolve the higher frequency modes of the system. This feature may prove particularly valuable…

Computational Physics · Physics 2015-05-13 Mathieu Drouin , Laurent Gremillet , Jean-Claude Adam , Anne Héron

A new spectral particle-in-cell (PIC) method for plasma modeling is presented and discussed. In the proposed scheme, the Fourier-Bessel transform is used to translate the Maxwell equations to the quasi-cylindrical spectral domain. In this…

Plasma Physics · Physics 2019-08-17 Igor A. Andriyash , Remi Lehe , Agustin Lifschitz

The development and the implementation of a Particle-in-Cell code written in the Unified Parallel C (UPC) language for plasma simulations with application to astrophysics and fusion nuclear energy machines are presented. A simple one…

Computational Physics · Physics 2011-08-10 Stefano Markidis , Giovanni Lapenta

In this paper we present a new three dimensional (3D) full electromagnetic relativistic hybrid plasma code H-VLPL (hybrid virtual laser plasma laboratory). The full kinetic particle-in-cell (PIC) method is used to simulate low density hot…

Computational Physics · Physics 2010-05-06 T. Tueckmantel , J. Liljo , A. Pukhov , M. Hochbruck

We present msmJAX, a Python package implementing the multilevel summation method with B-spline interpolation, a linear-scaling algorithm for efficiently evaluating electrostatic and other long-range interactions in particle-based…

Computational Physics · Physics 2025-10-08 Florian Buchner , Johannes Schörghuber , Nico Unglert , Jesús Carrete , Georg K. H. Madsen

We compare different Poisson solvers within the context of an electrostatic Vlasov-Poisson system. These schemes are implemented as part of the IPPL (Independent Parallel Particle Layer) library (Frey et al., 2024), which provides…

Computational Engineering, Finance, and Science · Computer Science 2026-05-08 Sonali Mayani , Paul Fischill , Sriramkrishnan Muralikrishnan , Andreas Adelmann

The particle-in-cell (PIC) method has been widely used for plasma simulation, because of its noise-reduction capability and moderate computational cost. The immersed finite element (IFE) method is efficient for solving interface problems on…

Numerical Analysis · Mathematics 2019-10-18 Jinwei Bai , Yong Cao , Yuchuan Chu , Xu Zhang

The particle-in-cell (PIC) method is successfully used to study magnetized plasmas. However, this requires large computational costs and limits simulations to short physical run-times and often to setups in less than three spatial…

High Energy Astrophysical Phenomena · Physics 2023-01-24 Rouven Lemmerz , Mohamad Shalaby , Timon Thomas , Christoph Pfrommer

We introduce JPC, a JAX library for training neural networks with Predictive Coding. JPC provides a simple, fast and flexible interface to train a variety of PC networks (PCNs) including discriminative, generative and hybrid models. Unlike…

Neural and Evolutionary Computing · Computer Science 2024-12-06 Francesco Innocenti , Paul Kinghorn , Will Yun-Farmbrough , Miguel De Llanza Varona , Ryan Singh , Christopher L. Buckley

We introduce Atomistic learned potentials in JAX (apax), a flexible and efficient open source software package for training and inference of machine-learned interatomic potentials. Built on the JAX framework, apax supports GPU acceleration…

Chemical Physics · Physics 2025-11-19 Moritz René Schäfer , Nico Segreto , Fabian Zills , Christian Holm , Johannes Kästner

While the particle-in-cell (PIC) method is quite mature, verification and validation of both newly developed methods and individual codes has largely focused on an idiosyncratic choice of a few test cases. Many of these test cases involve…

Computational Physics · Physics 2021-05-19 S. O'Connor , Z. D. Crawford , J. Verboncoeur , J. Lugisland , B. Shanker

The Particle-in-Cell (PIC) simulation has been a widely used method for studying plasma physics. However, fully three-dimensional PIC simulations always require huge computational resources. For problems with near azimuthal symmetry, recent…

Plasma Physics · Physics 2024-02-09 Jianzhao Wang , Weiming An , Rong Tang , Weiyu Meng , Jiayong Zhong

In this paper, Particle-in-Cell algorithms for the Vlasov-Poisson system are presented based on its Poisson bracket structure. The Poisson equation is solved by finite element methods, in which the appropriate finite element spaces are…

Numerical Analysis · Mathematics 2022-08-10 Anjiao Gu , Yang He , Yajuan Sun

We present jax-cosmo, a library for automatically differentiable cosmological theory calculations. It uses the JAX library, which has created a new coding ecosystem, especially in probabilistic programming. As well as batch acceleration,…

Cosmology and Nongalactic Astrophysics · Physics 2023-05-01 Jean-Eric Campagne , François Lanusse , Joe Zuntz , Alexandre Boucaud , Santiago Casas , Minas Karamanis , David Kirkby , Denise Lanzieri , Yin Li , Austin Peel

Basic principles of particle-in-cell (PIC ) codes with the main application for plasma-based acceleration are discussed. The ab initio full electromagnetic relativistic PIC codes provide the most reliable description of plasmas. Their…

Plasma Physics · Physics 2015-10-06 Alexander Pukhov

We propose the use of automatic differentiation through the programming framework jax for accelerating a variety of analysis tasks throughout gravitational wave (GW) science. Firstly, we demonstrate that complete waveforms which cover the…

Instrumentation and Methods for Astrophysics · Physics 2023-02-13 Thomas D. P. Edwards , Kaze W. K. Wong , Kelvin K. H. Lam , Adam Coogan , Daniel Foreman-Mackey , Maximiliano Isi , Aaron Zimmerman

We introduce $\texttt{A}$strophysical $\texttt{H}$ybrid-$\texttt{K}$inetic simulations with the $\texttt{flASH}$ code ($\texttt{AHKASH}$) -- a new Hybrid particle-in-cell (PIC) code developed within the framework of the multi-physics code…

Instrumentation and Methods for Astrophysics · Physics 2024-09-24 Radhika Achikanath Chirakkara , Christoph Federrath , Amit Seta

A fully variational, unstructured, electromagnetic particle-in-cell integrator is developed for integration of the Vlasov-Maxwell equations. Using the formalism of Discrete Exterior Calculus, the field solver, interpolation scheme and…

Plasma Physics · Physics 2015-06-18 Jonathan Squire , Hong Qin , William M. Tang

Particle-In-Cell codes are widely used for plasma physics simulations. It is often the case that particles within a computational cell need to be split to improve the statistics or, in the case of non-uniform meshes, to avoid the…

Computational Physics · Physics 2021-04-22 Roch Smets , Nicolas Aunais , ANdrea Ciardi , Matthieu Drouin , Martin Campos-Pino , Philip Deegan