Christopher Woolford
Physics PhD Student
Differentiable Particle in Cell Code
URL: https://github.com/uwplasma/PyPIC3D
Project Details
I am currently building a differentiable, 3D particle in cell code using the Jax library in Python to explore the use of automatic differentiation in basic plasma science and stellarator optimization.
Features
- Spectral and Finite Difference Solvers
- Electrostatic and Dynamic Methods
- Macroparticle Weight Scaling by Debye Length
- Just-in-time Compilation