mirror of
https://github.com/DifferentiableUniverseInitiative/JaxPM.git
synced 2025-02-23 01:57:10 +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.