Pinned Loading

1. TorchPQ TorchPQ Public

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

   Cuda 213 21

2. fast_pytorch_kmeans fast_pytorch_kmeans Public

   This is a pytorch implementation of k-means clustering algorithm

   Python 285 38

3. custom_matmul_kernels custom_matmul_kernels Public

   Customized matrix multiplication kernels

   Jupyter Notebook 53 6

4. gmm-torch gmm-torch Public

   Forked from ldeecke/gmm-torch

   Gaussian mixture models in PyTorch.

   Python 1 1

5. MinimapTracker-MobileNet MinimapTracker-MobileNet Public

   A League of Legends Minimap Tracker

   Python 9 2

6. DOKSparse DOKSparse Public

   sparse DOK tensors on GPU, pytorch

   Cuda 2