New `Trace_ELBO` that generalizes `Trace_ELBO`, `TraceEnum_ELBO`, and `TraceGraph_ELBO`#2893

Draft

ordabayevy wants to merge 62 commits into

pyro-ppl:devfrom

ordabayevy:fix-funsor-traceelbo

Commits on Jul 3, 2021

trace_elbo
Yerdos Ordabayev
committed
lint
Yerdos Ordabayev
committed

Commits on Jul 5, 2021

Merge branch 'dev' of https://github.com/pyro-ppl/pyro into fix-funsor-traceelbo
Yerdos Ordabayev
committed
test_gradient
Yerdos Ordabayev
committed

Commits on Jul 16, 2021

copy traceenum_elbo and add test model with poisson dist
Yerdos Ordabayev
committed
lint
Yerdos Ordabayev
committed

Commits on Jul 21, 2021

use constant funsor
Yerdos Ordabayev
committed

Commits on Jul 28, 2021

working version
Yerdos Ordabayev
committed
pass second test
Yerdos Ordabayev
committed

Commits on Jul 29, 2021

clean up trace_elbo
Yerdos Ordabayev
committed

Commits on Aug 8, 2021

add another test
Yerdos Ordabayev
committed

Commits on Aug 20, 2021

lazy eval
Yerdos Ordabayev
committed

Commits on Sep 18, 2021

Merge branch 'dev' of https://github.com/pyro-ppl/pyro into fix-funsor-traceelbo
Yerdos Ordabayev
committed

Commits on Sep 30, 2021

Merge branch 'dev' of https://github.com/pyro-ppl/pyro into fix-funsor-traceelbo
Yerdos Ordabayev
committed
vectorize particles; update tests
Yerdos Ordabayev
committed
minor fixes; pin to funsor@normalize-logaddexp
Yerdos Ordabayev
committed
update docs/requirements
Yerdos Ordabayev
committed
combine Trace_ELBO and TraceEnum_ELBO
Yerdos Ordabayev
committed

Commits on Oct 1, 2021

eager evaluation
Yerdos Ordabayev
committed
rm file
Yerdos Ordabayev
committed
lazy
Yerdos Ordabayev
committed
remove memoize
Yerdos Ordabayev
committed

Commits on Oct 10, 2021

merge TraceEnum_ELBO
Yerdos Ordabayev
committed

Commits on Oct 11, 2021

skip test
Yerdos Ordabayev
committed

Commits on Oct 12, 2021

fixes
Yerdos Ordabayev
committed
convert Tensor to Categorical
Yerdos Ordabayev
committed
restore docs/requirements.txt
Yerdos Ordabayev
committed
pin funsor in docs/requirements
Yerdos Ordabayev
committed
Merge branch 'dev' of https://github.com/pyro-ppl/pyro into fix-funsor-traceelbo
Yerdos Ordabayev
committed
use funsor.optimizer.apply_optimizer; higher precision in the test
Yerdos Ordabayev
committed
pin funsor to the latest commit
Yerdos Ordabayev
committed
optimize logzq
Yerdos Ordabayev
committed

Commits on Oct 13, 2021

optimize logzq
Yerdos Ordabayev
committed
restore TraceEnum_ELBO
Yerdos Ordabayev
committed
revert hmm changes
Yerdos Ordabayev
committed
_tensor_to_categorical helper function
Yerdos Ordabayev
committed
lazy to_funsor
Yerdos Ordabayev
committed
reduce over particle_var
Yerdos Ordabayev
committed
address comment in tests
Yerdos Ordabayev
committed
import pyroapi
Yerdos Ordabayev
committed

Commits on Oct 14, 2021

compute expected grads using dice factors
Yerdos Ordabayev
committed

Commits on Oct 15, 2021

add test with guide enumeration
Yerdos Ordabayev
committed
add two more tests
Yerdos Ordabayev
committed
pin funsor
Yerdos Ordabayev
committed
lint
Yerdos Ordabayev
committed
remove breakpoint
Yerdos Ordabayev
committed

Commits on Oct 29, 2021

Merge branch 'dev' of https://github.com/pyro-ppl/pyro into fix-funsor-traceelbo
Yerdos Ordabayev
committed

Commits on Nov 3, 2021

Approximate(ops.sample, ...) based approach
Yerdos Ordabayev
committed

Commits on Nov 4, 2021

Importance funsor based approach
Yerdos Ordabayev
committed
fixes
Yerdos Ordabayev
committed

Commits on Apr 6, 2022

Merge branch 'dev' into fix-funsor-traceelbo
Yerdos Ordabayev
committed

Commits on Apr 9, 2022

fix funsor model enumeration
Yerdos Ordabayev
committed
Merge branch 'fix-model-enumeration-funsor' into fix-funsor-traceelbo
Yerdos Ordabayev
committed

Commits on Apr 11, 2022

use Sampled funsor
Yerdos Ordabayev
committed
fixes
Yerdos Ordabayev
committed
git fixes
Yerdos Ordabayev
committed
Merge branch 'dev' into fix-funsor-traceelbo
Yerdos Ordabayev
committed
use Provenance funsor
Yerdos Ordabayev
committed

Commits on Apr 12, 2022

clean up
Yerdos Ordabayev
committed

Commits on May 5, 2022

fixes
Yerdos Ordabayev
committed
Merge branch 'dev' into fix-funsor-traceelbo
Yerdos Ordabayev
committed

Commits on Jun 22, 2022

use provenance
Yerdos Ordabayev
committed