It’s not converge according to the threshold you set (i.e., diff='absolute',tolerance=2e-4), however many reasons could cause this, you can also try increasing the number of mc to compute the gradient, and try different optimizer. More detail see the discussion in Poor Accuracy of ADVI for Linear Regression