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

Allow callable for vectorization over particles on SVI #1902

Merged

Conversation

juanitorduz
Copy link
Contributor

Closes #1645

@juanitorduz juanitorduz marked this pull request as draft November 10, 2024 10:29
numpyro/infer/elbo.py Outdated Show resolved Hide resolved
@juanitorduz juanitorduz force-pushed the allow-callable-svi-vectorization branch from 7228a78 to adb8ef2 Compare November 10, 2024 16:34
@juanitorduz
Copy link
Contributor Author

Thanks for the suggestion @fehiepsi ! I added it to the PR and added some tests.

@juanitorduz juanitorduz marked this pull request as ready for review November 10, 2024 18:27
@juanitorduz juanitorduz changed the title [WIP] Allow callable for vectorization over particles on SVI Allow callable for vectorization over particles on SVI Nov 10, 2024
@fehiepsi fehiepsi merged commit c8a0990 into pyro-ppl:master Nov 10, 2024
4 checks passed
@fehiepsi
Copy link
Member

Thanks, @juanitorduz!

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

Successfully merging this pull request may close these issues.

pmap over num_particles in SVI
2 participants