According to the docs, Bernoulli can take either p or logit_p as parameters, but not both. I chose to feed logit_p
ibernoulli = pm.Bernoulli.dist(logit_p=N_x_logit, shape=(N, T, 2))
But it kicks out with TypeError: init() missing 1 required positional argument: ‘p’
Is the doc wrong. Do you have to convert to p?