DifferentiableUniverseInitiative / JaxPM

JAX-powered Cosmological Particle-Mesh N-body Solver
37Updated last month

Alternatives and similar repositories for JaxPM:

Users that are interested in JaxPM are comparing it to the libraries listed below