David Majnemer

TPU-KNN: K Nearest Neighbor Search at Peak FLOP/s

2022 · arXiv (Cornell University)

This paper presents a novel nearest neighbor search algorithm achieving TPU (Google Tensor Processing Unit) peak performance, outperforming state-of-the-art GPU algorithms with similar level of recall. The design of the proposed algorithm is motivated by …