Hi! I am having the exact same problem. Have you solved it? @dustin has said a lot of times in this forum that ed.KLqp doesn’t work well with Gamma and Dirichlet variational approximations (eg: Error in inference.run() for a mixture model), but I never found an explanation for this.