Variational Inference for Gaussian Mixture models

This is indeed the issue!

I’ll fix this. Thanks!

EDIT: Should be fixed by Fix dimensionality issues of ADVI by torfjelde · Pull Request #2162 · TuringLang/Turing.jl · GitHub