Last released Aug 13, 2026
PM++: a JAX-based differentiable multi-GPU Particle-Mesh cosmology simulator.