You have to share more details about the model. numpyro running fast just means it’s getting a 100% divergences.