Graphical Model Components

thrml.AbstractNode

A node in a PGM.

Every node used in a PGM must inherit from this class. When compiling a program, each node is assigned a shape and datatype that are used to organize the state of the sampling program in a jax-friendly way.

thrml.SpinNode(thrml.AbstractNode)

A node that represents a random variable that takes on a state in {-1, 1}.

thrml.CategoricalNode(thrml.AbstractNode)

A node that represents a random variable that may take on any one of K possible discrete states, represented by a positive integer in (0, K].