Version dependant slowing down of Gaussian Mixture sampling in Ubuntu 20.04

[iavicenna]

@iavicenna

sigma = pm.HalfNormal('sigma', sigma=1, dims=['cluster', 'coord'])
    covs = [pt.diag(sigma[i]) for i in range(n_clusters)]

    # Define the mixture
    components = [pm.MvNormal.dist(mu=centroids[i], cov=covs[i]) for i in range(n_clusters)]

Thanks, I confirm, this way also gives the same speed up for 5.9.0. But 5.9.1 still stuck.