pyjuice
PC Creation
Construct a vector of input nodes defined on the same variable and have the same input distribution. |
|
Construct a vector of product nodes given a list of children PCs defined on disjoint sets of variables. |
|
Construct a vector of sum nodes given a list of children PCs defined on the same sets of variables. |
|
Context-manager that sets the block size of PC nodes constructed by inputs, multiply, and summate. |
PC Structure Transformation
Merge nodes with identical region node together. |
|
Generate an equivalent PC with potentially high block sizes. |
|
Decrease the block size of a PC. |
|
Create a deepcopy of the input PC. |
PC Compilation
Compile a PC represented by a DAG into an equivalent torch.nn.Module. |
IO
Load a PC from file. |
|
Save a PC to file. |