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

Log likelihood computation in numpyro can be extremely slow #2373

Open
OriolAbril opened this issue Aug 28, 2024 · 3 comments · May be fixed by #2390
Open

Log likelihood computation in numpyro can be extremely slow #2373

OriolAbril opened this issue Aug 28, 2024 · 3 comments · May be fixed by #2390

Comments

@OriolAbril
Copy link
Member

See #2367 for example. We should look into what makes it be slow and try to fix it, it would also be nice to add a progressbar so it is less surprising when things get stuck or freeze.
@aporsch1
Copy link

I think I'd like to work on this. I'll look at your guidelines and take a crack at it.

@OriolAbril
Copy link
Member Author

Thanks! That would be greatly appreciated. I myself am not very familiar with numpyro but will do my best to help if I can

@aporsch1
Copy link

Great. We aren't parallelizing at all here, is there any downside to doing that? We could include batch size as a parameter or something like that which could lead to a decent size speedup, I think. I will also look more at numpyro to see if I can get more performance gains there.

@OriolAbril OriolAbril linked a pull request Oct 7, 2024 that will close this issue
(WIP?) vectorized log_likelihood function for NumPyro #2390
Open
3 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

(WIP?) vectorized log_likelihood function for NumPyro aporsch1/arviz
2 participants
@OriolAbril @aporsch1