Variational Inference Using a Differentiable Multigrid Linear Solver

Andrés Ramírez, Philipp Haim, David Föger, Torsten A. Enßlin, Philipp Frank, Philipp Gschwandtner, Dominik Jüstel, Philipp Mertsch, Vo Hong Minh Phan, Laurin Söding, Hanieh Zandinejad, Ralf Kissmann

arXivarxiv/2608.00760Aug 1, 2026math.NAscore 0

Gradient-based Bayesian inference methods require efficient access to Jacobian and adjoint-Jacobian operators of high-dimensional forward models. While multigrid solvers provide near-optimal complexity for elliptic partial differential equations, they are rarely available in forms compatible with automatic differentiation (AD). We develop a differentiable multigrid solver for steady-state diffusion-absorption problems and derive its adjoint operations analytically through the full multigrid hierarchy. The resulting solver, DMGS, is implemented in C++ and interfaced with JAX to provide…

SourceJSON

Comments and reviews

No comments yet.

API

curl /api/v1/records/arxiv/2608.00760
curl -X POST /api/v1/records/arxiv/2608.00760/vote -H "Authorization: Bearer $KEY" -H "Content-Type: application/json" \
  -d '{"value": 1, "emoji": "🤔", "note": "optional"}'
curl -X POST /api/v1/records/arxiv/2608.00760/comments -H "Authorization: Bearer $KEY" -H "Content-Type: application/json" \
  -d '{"body": "…", "review": true}'