Skip to content

Pool emcee#66

Merged
marlena6 merged 8 commits into
mainfrom
pool_emcee
May 8, 2026
Merged

Pool emcee#66
marlena6 merged 8 commits into
mainfrom
pool_emcee

Conversation

@marlena6
Copy link
Copy Markdown
Collaborator

@marlena6 marlena6 commented May 3, 2026

Draft pull request to show how emcee with pool is working, it seems to work but messes with the previous Sampler structure so we should discuss how to make both parallelized and non-parallelized options work (two different sampler classes? Optional posterior argument to Sampler class? )

…t works in an interactive node but possibly not after submitted to batch. Needs further investigation. Also added notebook for check of windows in mocks.
@marlena6 marlena6 requested a review from andreufont May 3, 2026 21:02
@marlena6 marlena6 linked an issue May 3, 2026 that may be closed by this pull request
Copy link
Copy Markdown
Contributor

@andreufont andreufont left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Let's talk in person, I'm not sure I understand what is going on here.

@marlena6
Copy link
Copy Markdown
Collaborator Author

marlena6 commented May 4, 2026

New version doesn't involve the Sampler class anymore. Let me know if you'd like to update things to use a modified version of the class, or make a new class

@andreufont
Copy link
Copy Markdown
Contributor

Cool, that was fast! I'll take a look at it and try to run it myself. Thanks!

Comment thread scripts/mcmc_sampler.py
Comment thread scripts/mcmc_sampler.py
Comment thread scripts/mcmc_sampler.py
Comment thread scripts/mcmc_sampler.py
Copy link
Copy Markdown
Contributor

@andreufont andreufont left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think this could work, thanks @marlena6 !

I've added several comments to the code, but mostly for me to understand what we are doing. We can discuss it and try it a bit more before merging, but if you wanted we could also merge it now (it doesn't affect any other code)

@andreufont
Copy link
Copy Markdown
Contributor

@marlena6 - I was able to run this at NERSC. I'd still like to discuss the comments I made above, do you mind replying here on GitHub, or discussing in person next week?

After that I would merge this PR and open another issue about profiling the sampler at scale (check that the boost is larger than the factor of 5-6 that you mentioned earlier).

@andreufont andreufont marked this pull request as ready for review May 7, 2026 13:20
@marlena6 marlena6 merged commit f7020c6 into main May 8, 2026
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.

Timing for pool emcee Stop the sampling when the chains have converged Use Pool in the Sampler class

2 participants