I’m sure for some of you, this is super easy but for us novices, this seems to work.
with pm.model():
m = pm.Normal(“m”,mu=[0,1],sigma=[0,1])
c = pm.Uniform(“c”, lower = 0, upper=1)
param = pt.as_tensor([0,0,0])
param = pt.set_subtensor(param[0:2],m[0:2])
param = pt.set_subtensor(param[2],c)
Then you can pass param to a custom likelihood, without the broadcast problem.