src-d / kmcuda
Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA
☆794Updated last year
Related projects: ⓘ
- ☆706Updated last year
- ThunderSVM: A Fast SVM Library on GPUs and CPUs☆1,561Updated 5 months ago
- fast library for ANN search and KNN graph construction☆281Updated 6 years ago
- A CUDA implementation of the k-means clustering algorithm☆244Updated 12 years ago
- Metric learning algorithms in Python☆1,395Updated last month
- A library for k-nearest neighbor search☆377Updated 4 months ago
- FAst Lookups of Cosine and Other Nearest Neighbors (based on fast locality-sensitive hashing)☆1,133Updated 3 months ago
- ThunderGBM: Fast GBDTs and Random Forests on GPUs☆690Updated 7 months ago
- Training of Locally Optimized Product Quantization (LOPQ) models for approximate nearest neighbor search of high dimensional data in Pyth…☆561Updated 5 years ago
- ATen: A TENsor library for C++11☆677Updated 4 years ago
- Barnes-Hut t-SNE☆891Updated 3 years ago
- Python framework for fast (approximated) nearest neighbour search in large, high-dimensional data sets using different locality-sensitive…☆762Updated last year
- Fast k nearest neighbor search using GPU☆507Updated 6 years ago
- A lightweight library for PyTorch training tools and utilities☆1,658Updated this week
- ☆455Updated 4 years ago
- Python interface to GPU-powered libraries☆984Updated 11 months ago
- Deep Networks with Stochastic Depth☆473Updated 6 years ago
- Spatially-sparse convolutional networks. Allows processing of sparse 2, 3 and 4 dimensional data.Build CNNs on the square/cubic/hypercubi…☆404Updated 3 years ago
- Ladder network is a deep learning algorithm that combines supervised and unsupervised learning☆516Updated 7 years ago
- Open source implementation of "Spreading Vectors for Similarity Search"☆317Updated 3 years ago
- Python module for performing basic dense linear algebra computations on the GPU using CUDA.☆606Updated 4 years ago
- ☆1,655Updated 6 years ago
- Submanifold sparse convolutional networks☆2,030Updated 8 months ago
- Training RNNs as Fast as CNNs (https://arxiv.org/abs/1709.02755)☆2,104Updated 2 years ago
- Log TensorBoard events without touching TensorFlow☆630Updated last year
- Probabilistic Torch is library for deep generative models that extends PyTorch☆886Updated 4 months ago
- auto-tuning momentum SGD optimizer☆423Updated 6 years ago
- This is a Torch implementation of ["Deep Residual Learning for Image Recognition",Kaiming He, Xiangyu Zhang, Shaoqing Ren, Jian Sun](http…☆576Updated 4 years ago
- Navigating Spreading-out Graph For Approximate Nearest Neighbor Search☆621Updated 9 months ago
- Fast Randomized PCA/SVD☆491Updated 5 years ago