URL: https://github.com/uwplasma/PyPIC3D
I have built a differentiable, 3D particle in cell simulation in Python with the JAX library. I am currently extending this code to feature arbitrary coordinates and dynamical relativity for use in studying fully relativistic plasma phenomena.