Turbo-GMM: FlashAttention-Inspired GPU-Accelerated Gaussian Mixture Model

Momin Abbas, Sadia Asif, Asad Ul Haq +1

Expectation-maximization (EM) for Gaussian mixture models (GMMs) requires scoring $N$ data points against $K$ mixture components. Prior fused GPU implementations, notably Flash-GMM…