This PR revises our main tutorial to better demonstrate ChiRho's compatibility with tensor broadcasting and indexing counterfactual worlds using gather, IndexSet, and indices_of from chirho.indexed.ops.
In addition, this PR removes some lingering TODOs about making sure probabilistic inference is working as expected. I confirmed that SVI is working by adding a new visualization comparing ground_truth ATE to the posterior estimate. There are also some minor changes to phrasing throughout, see changes for detail.
This PR revises our main tutorial to better demonstrate ChiRho's compatibility with tensor broadcasting and indexing counterfactual worlds using
gather
,IndexSet
, andindices_of
fromchirho.indexed.ops
.In addition, this PR removes some lingering TODOs about making sure probabilistic inference is working as expected. I confirmed that SVI is working by adding a new visualization comparing ground_truth ATE to the posterior estimate. There are also some minor changes to phrasing throughout, see changes for detail.