mirror of
https://github.com/DifferentiableUniverseInitiative/JaxPM.git
synced 2025-02-22 17:47:11 +00:00
JAX-powered Cosmological Particle-Mesh N-body Solver
.gitignore | ||
design.md | ||
LICENSE | ||
README.md |
JaxPM
JAX-powered Cosmological Particle-Mesh N-body Solver
The goal of this project is to build the next generation of differentiable PM solvers, using JAX as the backend.
This project is currently in an early design phased. All inputs are welcome on the design document