python - GPU 上の BIC または AIC を使用したガウス混合モデル

Question

scikit-learn には GMM を使用した bic/aic 基準がありますが、データを GPU に適合させたいと考えています。CuPy(cuda numpy wrapper)に実装されたGMMを見つけましたが、bic/aic基準がありません.( https://github.com/cupy/cupy/blob/master/examples/gmm/gmm.py )

そのコードに bic/aic 基準を実装する方法は? または良いライブラリはありますか？助けてください。