Multivariate categorical with observed data

This may be of help: Distribution Dimensionality — PyMC 5.3.0 documentation