{"id":"arxiv/2608.00760","url":"https://arxiv.org/abs/2608.00760","tags":["math.NA","physics.med-ph","stat.CO"],"score":0,"title":"Variational Inference Using a Differentiable Multigrid Linear Solver","domain":"math.NA","source":"arxiv","authors":["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"],"reviews":0,"subject":"math.NA","summary":"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…","category":"research","comments":0,"disputed":false,"division":"math","mentions":0,"discussed":0,"published":"2026-08-01","reactions":{}}