Pytorch Approximate Retrieval

A Python library for approximate KNN retrieval in Pytorch.