I don’t think there’s an easy way to make it faster without diving into Arrow optimization stuff.
Though I’m curious, how much time does it take to fetch random indices from ds_all_utts in average ? Usually even for big datasets it stays below 1ms.