Numpyro sampler but do not return InferenceObject

1

Hi to anyone who can help me.

I am tying to user the numpyro sampler as it is much faster for the model im trying to sample from. I’m basing my work of some old pymc3 code, and it is why i specify return_inferencedata=False so i can make use of already built plotting utilities. But an Inference object is returned. Is it simple not possible when using the numpyro sampler?

with mod:
trace = pm.sample(nuts_sampler=“numpyro”,draws=2000, tune=2000, chains=4, cores=8, target_accept=0.95, return_inferencedata=False)

2

The way forward is InferenceData so we didn’t offer the old MultiTrace for new samplers, to also reduce code complexity. Is it too much work to update the plotting functionality?

3

It will be, but it will be worth updating. The speed up in sampling makes it worth it. Thank you for the quick response!

1 Like