We are getting the initilisation error for 3 different models with the same data. We had to take some independent variables out that start with 0 and then have positives. Does numpyro ever only initialise on the top N rows of data?
Hi @dirknbr could you post an example model, even a simplified version?
I think it was due to transformations that lead to na. f(x) → na