Sounds like you are describing a latent dirichlet allocation (LDA): https://docs.pymc.io/notebooks/lda-advi-aevb.html