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 ffb82c3
Show file tree
Hide file tree
Showing 2 changed files with 500 additions and 73 deletions.
Loading

0 comments on commit ffb82c3

Please sign in to comment.