I\'m building up a mixture of models in tensorflow-probability. The joint distribution for one given model is:
one_network_prior = tfd.JointDistributionNamed(