search

erikbern / ann-benchmarks

Benchmarks of approximate nearest neighbor libraries in Python
http://ann-benchmarks.com
MIT License
4.99k stars 747 forks source link

Evaluate TorchPQ #294

Open ljak opened 2 years ago

ljak commented 2 years ago

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!

erikbern commented 2 years ago

interesting – yeah seems like it would be a good addition

maumueller commented 2 years ago

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.