TPU-KNN: K Nearest Neighbor Search at Peak Flop/S
arxiv.org