_images/JAX-in-Cell_logo.png GitHub repository

JAX-in-Cell DocumentationΒΆ

Welcome to JAX-in-Cell, an open-source JAX-based Particle-in-Cell (PIC) framework for simulating plasmas with GPU/TPU acceleration, autodifferentiation, and JIT compilation.