GPU上具有BIC或AIC的高斯混合模型

Question

在scikit-learn中有GMM的bic / aic准则，但是我想在GPU上适应我的数据。 我发现GMM在CuPy（cuda numpy包装器）中实现，但是没有bic / aic准则。（https://github.com/cupy/cupy/blob/master/examples/gmm/gmm.py）

如何对该代码实施bic / aic标准？还是有好的图书馆？请帮忙。