PyPIC3D is a 3D particle-in-cell (PIC) simulation code developed in Python, leveraging the Jax library for high-performance numerical computing. The primary goal of this project is to explore the feasibility of using Python and Jax for plasma simulations, which are traditionally performed using languages like C++ and Fortran.
- Autodifferentiation: Jax's built-in autodifferentiation capabilities allow for efficient gradient computations, which can be highly beneficial for optimization and surrogate modeling in plasma simulations.
- Benchmarking: The code is benchmarked against other open source PIC codes including OSIRIS.
- Plasma Simulation: Designed to simulate plasma behavior in three dimensions, providing insights into complex plasma dynamics.
PyPIC3D can be used for various plasma simulation applications, including but not limited to:
- Stellarator Simulations: Investigating the behavior of plasma in stellarator devices, which are used in fusion research.
- Optimization: Utilizing autodifferentiation for optimizing plasma configurations and parameters.
- Surrogate Modeling: Creating surrogate models to approximate plasma behavior, reducing computational costs for large-scale simulations.
PyPIC3D can be install from the PyPI repository using PIP by running the following command:
pip install PyPIC3DPyPIC3D can also be installed from source by cloning the repository and running the following:
cd PyPIC3D
pip install .To get started with PyPIC3D, follow the instructions in the read the docs to create a configuration file or use an existing demo.
To run PyPIC3D on a configuration file, run the following command:
PyPIC3D --config filename.tomlThe PyPIC3D codebase is organized into several key sections:
- Core: Contains the core functionalities of the PIC simulation, including particle push, field solve, and boundary conditions.
- Experiments: Contains several base experiment classes to do various computational experiments such as parameter scans.
- Demos: Demos of different plasma phenomenon that you can simulate using PyPIC3D.
- Tests: Unit tests and integration tests to ensure the correctness and stability of the codebase.
Each section is designed to be modular and easily extensible, allowing users to customize and expand the functionality of PyPIC3D according to their needs.
Contributions to PyPIC3D are welcome. Please refer to the contributing guidelines in the repository for more information on how to contribute.
This project is licensed under the MIT License. See the LICENSE file for details.