src-d / kmcuda
Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA
☆811Updated 2 years ago
Alternatives and similar repositories for kmcuda:
Users that are interested in kmcuda are comparing it to the libraries listed below
- A CUDA implementation of the k-means clustering algorithm☆249Updated 12 years ago
- GPU Accelerated t-SNE for CUDA with Python bindings☆1,821Updated 3 months ago
- FAst Lookups of Cosine and Other Nearest Neighbors (based on fast locality-sensitive hashing)☆1,147Updated 7 months ago
- fast library for ANN search and KNN graph construction☆288Updated 7 years ago
- Training of Locally Optimized Product Quantization (LOPQ) models for approximate nearest neighbor search of high dimensional data in Pyth…☆564Updated 5 years ago
- ThunderSVM: A Fast SVM Library on GPUs and CPUs☆1,581Updated 9 months ago
- Barnes-Hut t-SNE☆899Updated 3 years ago
- ☆712Updated last year
- Log TensorBoard events without touching TensorFlow☆630Updated 2 years ago
- Make huge neural nets fit in memory☆2,749Updated 4 years ago
- Simple package that makes your generator work in background thread☆277Updated 2 years ago
- Fast EMD for Python: a wrapper for Pele and Werman's C++ implementation of the Earth Mover's Distance metric☆484Updated last year
- Open source implementation of "Spreading Vectors for Similarity Search"☆319Updated 3 years ago
- Memory consumption and FLOP count estimates for convnets☆918Updated 6 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,034Updated last year
- Standalone TensorBoard for visualizing in deep learning☆371Updated 4 years ago
- A library for k-nearest neighbor search☆383Updated 9 months ago
- A simple memory manager for CUDA designed to help Deep Learning frameworks manage memory☆296Updated 6 years ago
- PyTorch Extension Library of Optimized Graph Cluster Algorithms☆844Updated 2 weeks ago
- Spatially-sparse convolutional networks. Allows processing of sparse 2, 3 and 4 dimensional data.Build CNNs on the square/cubic/hypercubi…☆406Updated 3 years ago
- Profiling and inspecting memory in pytorch☆1,040Updated 5 months ago
- Python framework for fast (approximated) nearest neighbour search in large, high-dimensional data sets using different locality-sensitive…☆766Updated last year
- Training RNNs as Fast as CNNs (https://arxiv.org/abs/1709.02755)☆2,101Updated 3 years ago
- Distributed TensorFlow basics and examples of training algorithms☆643Updated 6 years ago
- Fast K-Nearest Neighbor search with GPU☆142Updated 7 years ago
- Code for: "And the bit goes down: Revisiting the quantization of neural networks"☆634Updated 4 years ago
- TVM integration into PyTorch☆453Updated 5 years ago
- A lightweight library for PyTorch training tools and utilities☆1,682Updated this week
- A curated list of awesome architecture search resources☆1,186Updated 4 years ago
- ATen: A TENsor library for C++11☆690Updated 5 years ago