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