pytorch_gmm
