Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Should pyrenew's default MCMC sampler backend be numpyro's bundled NUTS sampler? #466

Open
dylanhmorris opened this issue Oct 1, 2024 · 3 comments

Comments

@dylanhmorris
Copy link
Collaborator

Or should the standard pattern be to specify the model in numpyro but pass it off to bayeux or blackjax etc for inference?

@CDCgov/pyrenew-devs

See also #361 #151

@dylanhmorris dylanhmorris changed the title Should pyrenew's sampler backend be numpyro's bundled NUTS sampler Should pyrenew's default MCMC sampler backend be numpyro's bundled NUTS sampler? Oct 1, 2024
@damonbayer
Copy link
Collaborator

I've been scared away from bayeux a bit. Seems like it is not exactly plug and play with numpyro: jax-ml/bayeux#51

@AFg6K7h4fhy2
Copy link
Collaborator

My preference stands with numpyro over blackjax. Not sure where I am with bayeux given that I have not done much with it. Having other JAX based sampling options available seems good, but for the default I presently think deferring to numpyro seems best (this could ofc change if I learned bayeux was superior in some manner).

@seabbs
Copy link
Collaborator

seabbs commented Oct 1, 2024

I did a bunch of digging into within chain parallelization. My current takeaway is that jax can do it and maybe some of the other backends can but numpyro can't. If that actually is the case it puts more weight on being backend agnostic at least sometime in the future IMO.

I really can't see how or why that would be the case though so very keen to see evidence to the contrary from people who have actually fit a numpyro model and so might be better informed.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants