I have a (complicated) Poisson model that perfectly works when I don’t have an offset variable (e.g. y=exp(a+Bx)). On the contrary, when I specify an offset (y = exp(a+Bx+logE)) this is returned (at around 3% of the NUTS using jitter+adapt_diag process):
ValueError: Bad initial energy: nan. The model might be misspecified.
How can I debug it? I tried to print the test_points but I don’t see anything strange