this might just be numpy indexing with `np.triu_indices`