JAX-powered Cosmological Particle-Mesh N-body Solver
Find a file
2021-07-17 01:02:55 +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 01:02:55 +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.

This project is currently in an early design phased. All inputs are welcome on the design document