Mixture Model Latent Variable (augmented-data) example

Does pymc.Mixture — PyMC 5.6.1 documentation do what you need?