perf: optimize segment index extraction using np.searchsorted#2

KhudaBuxMagsi wants to merge 1 commit into

google-deepmind:maingoogle-deepmind/seeing_without_pixels:mainfrom

KhudaBuxMagsi:optimize-segment-extractionKhudaBuxMagsi/seeing_without_pixels:optimize-segment-extractionCopy head branch name to clipboard