Skip to content

New ADVI API - #8

Closed
jessegrabowski wants to merge 3 commits into
mainfrom
advi-refactor
Closed

New ADVI API#8
jessegrabowski wants to merge 3 commits into
mainfrom
advi-refactor

Conversation

@jessegrabowski

Copy link
Copy Markdown
Owner

This PR moves the work from pymc-devs/pymc#7799, with the guide model stuff. I also added:

  • stick the landing estimator
  • forward sampling helper
  • LLM sketch for a training API patterned after pytorch-lightning

I updated the notebook to use the proposed training API. It obviously needs a lot of work (being that it's llm trash) but hopefully it can get some ideas flowing.

@review-notebook-app

Copy link
Copy Markdown

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

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.

1 participant