bartz.debug.TraceWithOffset

class bartz.debug.TraceWithOffset(leaf_tree, var_tree, split_tree, offset)

A trace of trees with an offset, compatible with bartz.mcmcloop.evaluate_trace.

has_chains: ClassVar[bool] = False

No chain axis; each leading axis is just the sample axis.

mesh: ClassVar[Mesh | None] = None

No device mesh; the trees are host-built and unsharded.

classmethod from_trees_trace(trees, offset)

Create a TraceWithOffset from a TreeHeaps.

Return type:

TraceWithOffset