Skip to content

feat: add numpyro checkpointing#103

Open
davecwright3 wants to merge 1 commit intonanograv:mainfrom
davecwright3:feat/numpyro-resume
Open

feat: add numpyro checkpointing#103
davecwright3 wants to merge 1 commit intonanograv:mainfrom
davecwright3:feat/numpyro-resume

Conversation

@davecwright3
Copy link
Collaborator

  • Adds checkpointing to NumPyro sampler
  • Allows users to resume sampling from a saved sampler state

@davecwright3 davecwright3 requested a review from vallis September 5, 2025 21:20
@davecwright3

This comment was marked as off-topic.

also allows users to resume sampling
@davecwright3
Copy link
Collaborator Author

rebased onto main

@meyers-academic meyers-academic self-assigned this Feb 13, 2026
@meyers-academic meyers-academic requested review from meyers-academic and removed request for vallis February 13, 2026 14:40
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.

2 participants