Skip to content

Conversation

fritzo
Copy link
Member

@fritzo fritzo commented Sep 9, 2021

Addresses #2813

This implements a new method of constructing autoguides, AutoWrap, that allow guides to leverage deterministic computations in the model, e.g. those introduced by reparameterization.

Tasks

  • mask model sites in the guide?
  • return deterministic sites in get_dependencies()
  • trace all sites in ._setup_prototype()
  • replace matrix-vector ops with einsum ops
  • avoid re-running the model in SVI
  • avoid re-running the model in Predictive

Tested

  • add to test_autoguide.py
  • test structure inference
  • test on a real world model

@fritzo fritzo added the WIP label Sep 9, 2021
@fritzo fritzo marked this pull request as draft September 9, 2021 23:04
@fritzo fritzo mentioned this pull request Sep 13, 2021
7 tasks
@fritzo
Copy link
Member Author

fritzo commented Sep 14, 2021

Closing in favor of #2929

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant