funsor icon indicating copy to clipboard operation
funsor copied to clipboard

Provenance funsor

Open ordabayevy opened this issue 3 years ago • 1 comments

Provenance tracking implementation in funsor. Here the provenance is the set of (name, point) tuples of RV samples that were in the history of the computations of the tracked term. Substitution results in the multiplication by the density:

>>> logp_x = Provenance(logp_value, {("x", point)})
>>> logp_x(x=point)
logp_value

The Provenance funsor is intended to be used in https://github.com/pyro-ppl/pyro/pull/2893 as a wrapper for ProvenanceTensor.

ordabayevy avatar Apr 11 '22 21:04 ordabayevy

Tests are failing due to the new jax version #594

ordabayevy avatar May 05 '22 16:05 ordabayevy