src-d / kmcuda
Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA
☆826Updated 2 years ago
Alternatives and similar repositories for kmcuda:
Users that are interested in kmcuda are comparing it to the libraries listed below
- Fast k nearest neighbor search using GPU☆522Updated 6 years ago
- fast library for ANN search and KNN graph construction☆288Updated 7 years ago
- A library for k-nearest neighbor search☆387Updated 11 months ago
- ATen: A TENsor library for C++11☆694Updated 5 years ago
- Fast EMD for Python: a wrapper for Pele and Werman's C++ implementation of the Earth Mover's Distance metric☆485Updated 2 years ago
- Barnes-Hut t-SNE☆898Updated 3 years ago
- A CUDA implementation of the k-means clustering algorithm☆251Updated 12 years ago
- PyTorch Extension Library of Optimized Graph Cluster Algorithms☆859Updated 2 months ago
- Fast K-Nearest Neighbor search with GPU☆141Updated 7 years ago
- Open source implementation of "Spreading Vectors for Similarity Search"☆319Updated 3 years ago
- Python framework for fast (approximated) nearest neighbour search in large, high-dimensional data sets using different locality-sensitive…☆767Updated 2 years ago
- Python interface to GPU-powered libraries☆991Updated last year
- Training of Locally Optimized Product Quantization (LOPQ) models for approximate nearest neighbor search of high dimensional data in Pyth…☆565Updated 5 years ago
- Parallel t-SNE implementation with Python and Torch wrappers.☆1,900Updated last year
- an Extremely Fast Approximate Nearest Neighbor graph construction Algorithm framework☆93Updated 5 years ago
- Tutorial for building a custom CUDA function for Pytorch☆512Updated 6 years ago
- ThunderGBM: Fast GBDTs and Random Forests on GPUs☆696Updated last week
- Navigating Spreading-out Graph For Approximate Nearest Neighbor Search☆667Updated last year
- Training RNNs as Fast as CNNs (https://arxiv.org/abs/1709.02755)☆2,099Updated 3 years ago
- Neural network visualizer☆956Updated 4 years ago
- A Python module for getting the GPU status from NVIDA GPUs using nvidia-smi programmically in Python☆1,175Updated 11 months ago
- ThunderSVM: A Fast SVM Library on GPUs and CPUs☆1,590Updated last year
- ☆713Updated last year
- CuPy fused PyTorch neural networks ops☆273Updated 7 years ago
- kmeans clustering with multi-GPU capabilities☆120Updated last year
- Weighted MinHash implementation on CUDA (multi-gpu).☆117Updated last year
- Log TensorBoard events without touching TensorFlow☆627Updated 2 years ago
- Matrix Shadow:Lightweight CPU/GPU Matrix and Tensor Template Library in C++/CUDA for (Deep) Machine Learning☆1,110Updated 5 years ago
- TVM integration into PyTorch☆452Updated 5 years ago
- Standalone TensorBoard for visualizing in deep learning☆369Updated 5 years ago