COSMIT: Minor opt in pairwise_distances_argmin_min