Custom multivariate density via DensityDist

This guide may help you: Distribution Dimensionality — PyMC dev documentation. You can use the same tools to reason about shapes of the density dist.

Edit: Original link was wrong