normalize_cols function

Build an optree pipeline that normalizes a set of columns so each column sums to one in each partition.