Projects
fast_pytorch_kmeans
This is a pytorch implementation of k-means clustering algorithm
Python - Released: 14 Sep 2020 - 280
TorchPQ
Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda
Cuda - Released: 08 Feb 2021 - 211
custom_matmul_kernels
Customized matrix multiplication kernels
Jupyter Notebook - Released: 02 May 2021 - 52