TPU-KNN: K Nearest Neighbor Search at Peak FLOP/s - Chern, Felix; Hechtman, Blake; Davis, Andy; Guo, Ruiqi; Majnemer, David; Kumar, Sanjiv (Advances in Neural Information Processing Systems, 2022)