Extend partial_sum_product() to cases tractable for Gaussians - Githubissues

pyro-ppl / funsor

Functional tensors for probabilistic programming

https://funsor.pyro.ai

Apache License 2.0

236 stars 20 forks source link

Extend partial_sum_product() to cases tractable for Gaussians #584

Closed fritzo closed 2 years ago

fritzo commented 2 years ago

Addresses #559 ~~Blocked by #583~~ pair coded with @eb8680

Tested

added tests of intractable cases
added tests for unreduced sum variables in plates
added tests for batching, i.e. unreduced plate variables