Papers citing 'BlackJAX: Composable Bayesian inference in JAX'

Title
CLAX: Fast and Flexible Neural Click Models in JAX Philipp Hager O. Zoeter Maarten de Rijke 60 0 0 05 Nov 2025
Reducing normalizing flow complexity for MCMC preconditioning David Nabergoj Erik Štrumbelj 92 0 0 04 Nov 2025
Fisher meets Feynman: score-based variational inference with a product of experts Diana Cai Robert Mansel Gower David M. Blei Lawrence K. Saul 85 0 0 24 Oct 2025
Efficiently Vectorized MCMC on Modern Accelerators Hugh Dance Pierre Glaser Peter Orbanz Ryan P. Adams 215 1 0 20 Mar 2025
Geodesic Slice Sampler for Multimodal Distributions with Strong CurvatureConference on Uncertainty in Artificial Intelligence (UAI), 2025 Bernardo Williams Hanlin Yu Hoang Phuc Hau Luu Georgios Arvanitidis Arto Klami 249 0 0 28 Feb 2025
Simulation-based inference with the Python Package sbijax Simon Dirmeier S. Ulzega Antonietta Mira Carlo Albert 204 5 0 28 Sep 2024
Diffusion Generative Modelling for Divide-and-Conquer MCMC C. Trojan Paul Fearnhead C. Nemeth DiffM 149 1 0 17 Jun 2024