docs: def marginal(repertoire, node_index):
      """Get the marginal distribution for a node."""
      index = tuple(i for i in range(repertoire.ndim) if i != node_index)
  
      return repertoire.sum(index, keepdims=True)