Hi there, I tried to rerun this Pathfinder Variational Inference: [image] Pathfinder Variational Inference Pathfinder[ Zhang et al., 2021] is a variational inference algorithm that produces samples from the posterior of a Bayesian model. It compares favorably to the widely used ADVI alg…

Hi @Dennis_Ma , Best to use the pymc backend for Pathfinder for now. There’s still a few more developments under way for blackjax. The pymc backend for Pathfinder is fully functional–with Multi-Path Pathfinder available! [image] Dennis_Ma: pmx.fit(num_paths=16,num_draws_per_path=1000, num_draws…

Pathfinder Variational Inference Not working on GPU (WSL)

Questions v5

Dennis_Ma February 20, 2025, 6:28pm 5

Thank you so much for the detailed explanation!! Looking forward to the new Pathfinder example notebook!

1 Like

Topic		Replies	Views
Does JAX backend Speedup also apply to variational inference? v5	2	761	January 20, 2023
Did the Pathfinder algorithm ever get implemented in PyMC sampling	3	624	February 13, 2022
PyMC3 v3.2 release News	0	880	October 18, 2017
Leveraging the GPU in PyMC3 Questions gpu , theano	5	8052	August 12, 2017
`sample_node` no longer working? Questions	10	763	September 11, 2017

Pathfinder Variational Inference Not working on GPU (WSL)

Related topics