Skip to content

Commit

Permalink
Support more kinds of marginalization via dim analysis
Browse files Browse the repository at this point in the history
This commit lifts the restriction that only Elemwise operations may link marginalized to dependent RVs. We map input dims to output dims, to assess whether an operation mixes information from different dims or not. Graphs where information is not mixed can be efficiently marginalized.
  • Loading branch information
ricardoV94 committed Feb 16, 2024
1 parent 79d976d commit 2aa1e81
Show file tree
Hide file tree
Showing 2 changed files with 510 additions and 72 deletions.
Loading

0 comments on commit 2aa1e81

Please sign in to comment.