Gaussian Mixture Model Pytorch Cuda 12