Add blackjax sampler to sampling_jax.py #5454

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.

Sign up for GitHub

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

Jump to bottom

Closed

twiecki opened this issue Feb 9, 2022 · 1 comment

Labels

beginner friendly help wanted jax v4

Member

twiecki commented Feb 9, 2022

There is now a functional implementation of running the blackjax sampler on PyMC: https://github.com/blackjax-devs/blackjax/blob/main/examples/use_with_pymc3.ipynb

It should be straight-forward to just add a wrapper for it just like we have for the numpyro sampler.

twiecki added beginner friendly help wanted jax v4 labels

fonnesbeck self-assigned this

twiecki mentioned this issue

Adding NUTS sampler from blackjax to sampling_jax #5477

Merged

2 tasks

twiecki unassigned fonnesbeck

Member

ricardoV94 commented Mar 19, 2022

Closed via #5477

ricardoV94 closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment