mirror of
https://github.com/DifferentiableUniverseInitiative/JaxPM.git
synced 2025-02-22 17:47:11 +00:00
176 B
176 B
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.