JAX-in-Cell: A Differentiable Particle-in-Cell Code for Plasma Physics Applications

Longyu Ma, Rogerio Jorge, Hongke Lu +2

JAX-in-Cell is a fully electromagnetic, multispecies, and relativistic 1D3V Particle-in-Cell (PIC) framework implemented entirely in JAX. It provides a modern, Python-based alterna…