Pinned Loading

1. TorchPQ TorchPQ Public

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

   Cuda 204 19

2. fast_pytorch_kmeans fast_pytorch_kmeans Public

   This is a pytorch implementation of k-means clustering algorithm

   Python 264 34

3. custom_matmul_kernels custom_matmul_kernels Public

   Customized matrix multiplication kernels

   Jupyter Notebook 52 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