I would like to use multiple chains for ADVI. So far, i could not find a single documentation on how to sample multiple chains in ADVI?
If multiple chains are not possible, i would like to append each run of pm.variational.approximations.sample_approx (…) such that i get a final trace of type
“<MultiTrace: ‘n’ chains, ‘x’ iterations, ‘m’ variables>”
from appending ‘n’ samples of
“<MultiTrace: ‘1’ chains, ‘x’ iterations, ‘m’ variables>”
Multiple chains are typically used to check MCMC convergence criteria. These criteria do not make sense for ADVI, as an approximating Gaussian is fit in closed form. The sampling you get from ADVI does not involve MCMC at all – and so there’s no such thing as a “chain” for ADVI – which is why there is no documentation on the subject.