gpu accelerated implementation of k-NN using PyTorch. - View it on GitHub