bartz.debug.MinimalTrace¶

class bartz.debug.MinimalTrace(leaf_tree, var_tree, split_tree, offset, leaf_unit)[source]¶

A minimal trace of trees, compatible with bartz.mcmcloop.evaluate_trace.

offset: Float32[Array, '']¶: Constant shift added to the scaled sum of trees.

leaf_unit: Float32[Array, '']¶: The storage unit of the leaf values, 1 for leaves in data units.

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.