https://hgpu.org/?p=26973
TPU-KNN: K Nearest Neighbor Search at Peak FLOP/s