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.

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

Add JAX implementation for BinomialRV #1321

Open
Tracked by #1425
rlouf opened this issue Dec 3, 2022 · 2 comments
Open
Tracked by #1425

Add JAX implementation for BinomialRV #1321

rlouf opened this issue Dec 3, 2022 · 2 comments
Labels
enhancement New feature or request good first issue Good for newcomers help wanted Extra attention is needed JAX Involves JAX transpilation random variables Involves random variables and/or sampling

Comments

@rlouf
Copy link
Member

rlouf commented Dec 3, 2022

No description provided.

@rlouf rlouf added JAX Involves JAX transpilation enhancement New feature or request good first issue Good for newcomers help wanted Extra attention is needed random variables Involves random variables and/or sampling labels Dec 3, 2022
@rlouf rlouf mentioned this issue Feb 13, 2023
Open
@FredericWantiez
Copy link
Contributor

FredericWantiez commented Feb 14, 2023 •
edited
Loading

Can I give this one a try ? Made some progress here (it doesn't quite work yet).
It's mostly a few tweaks on: https://doi.org/10.1145/42372.42381 and the Numpyro implementation. Seems like they already made all of the heavy lifting for a Jax BinomialRV

@brandonwillard
Copy link
Member

Can I give this one a try ? Made some progress here (it doesn't quite work yet). It's mostly a few tweaks on: https://doi.org/10.1145/42372.42381 and the Numpyro implementation. Seems like they already made all of the heavy lifting for a Jax BinomialRV

That would be awesome, and greatly appreciated!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
enhancement New feature or request good first issue Good for newcomers help wanted Extra attention is needed JAX Involves JAX transpilation random variables Involves random variables and/or sampling
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@brandonwillard @rlouf @FredericWantiez