How do I run the Numpyro model line by line in order to debug it?

In Pyro, we can run the model line by line to verify that the model is running at each step. How do I do this in Numpyro :thinking:? Thanks a lot!

I guess you need

with numpyro.handlers.seed(rng_seed=0):
    # do line by line statements  
2 Likes

Thanks a loat, that’s exactly what I need