Warm-started Chebyshev geometric multigrid is competitive with distributed FFTs for FastPM and enables a differentiable moving-mesh particle–mesh gravity solver in JAX.
Title resolution pending
1 Pith paper cite this work. Polarity classification is still indexing.
1
Pith paper citing it
fields
astro-ph.IM 1years
2026 1verdicts
CONDITIONAL 1representative citing papers
citing papers explorer
-
Fast(er)PM and Moving Mesh: JAX-native Geometric Multigrid Methods
Warm-started Chebyshev geometric multigrid is competitive with distributed FFTs for FastPM and enables a differentiable moving-mesh particle–mesh gravity solver in JAX.