python - 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 标准？或者有好的图书馆吗？请帮忙。