Numpyro sampling warning: "Constant folding an instruction is taking > 1s:" Should I worry?

Thanks a lot for the quick reply!