JAX-powered Cosmological Particle-Mesh N-body Solver
Find a file
2021-07-17 00:59:03 +02:00
.gitignore Initial commit 2021-07-16 23:10:42 +02:00
design.md Update design.md 2021-07-17 00:59:03 +02:00
LICENSE Initial commit 2021-07-16 23:10:42 +02:00
README.md Update README.md 2021-07-17 00:31:44 +02:00

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.