/TorchPQ

Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda

Primary LanguageCudaMIT LicenseMIT

Stargazers