ann-benchmarks icon indicating copy to clipboard operation
ann-benchmarks copied to clipboard

Published 20 hours ago • verified publisher icon erikbern

Evaluate TorchPQ

Open ljak opened this issue 3 years ago • 2 comments

Hi,

Thanks for the great work !

I found out TorchPQ recently : https://github.com/DeMoriarty/TorchPQ

TorchPQ is a python library for Approximate Nearest Neighbor Search (ANNS) and Maximum Inner Product Search (MIPS) on GPU using Product Quantization (PQ) algorithm. TorchPQ is implemented mainly with PyTorch, with some extra CUDA kernels to accelerate clustering, indexing and searching.

It could be nice to be added to the list (I'm not the author).

Thanks!

ljak avatar Feb 10 '22 19:02 ljak

interesting – yeah seems like it would be a good addition

erikbern avatar Feb 10 '22 19:02 erikbern

Seems indeed interesting, but our current version doesn't work nicely with GPUs. It might be a bit difficult to get this included and requires some changes in our docker setup.

If you have the resources @ljak, it would be great if you could give it a try. I'd be happy to help.

maumueller avatar Feb 11 '22 08:02 maumueller