Alternative method for meanfield is commented out, since it yields worse KL divergence