src-d / kmcudaLinks
Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA
☆835Updated 2 years ago
Alternatives and similar repositories for kmcuda
Users that are interested in kmcuda are comparing it to the libraries listed below
Sorting:
- ATen: A TENsor library for C++11☆703Updated 5 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,049Updated 2 years ago
- fast library for ANN search and KNN graph construction☆295Updated 7 years ago
- A CUDA implementation of the k-means clustering algorithm☆252Updated 13 years ago
- Training of Locally Optimized Product Quantization (LOPQ) models for approximate nearest neighbor search of high dimensional data in Pyth…☆567Updated 6 years ago
- ☆715Updated 2 years ago
- ThunderSVM: A Fast SVM Library on GPUs and CPUs☆1,612Updated last year
- FAst Lookups of Cosine and Other Nearest Neighbors (based on fast locality-sensitive hashing)☆1,153Updated last year
- Barnes-Hut t-SNE☆909Updated 4 years ago
- Spatially-sparse convolutional networks. Allows processing of sparse 2, 3 and 4 dimensional data.Build CNNs on the square/cubic/hypercubi…☆407Updated 4 years ago
- Tutorial for building a custom CUDA function for Pytorch☆520Updated 6 years ago
- ThunderGBM: Fast GBDTs and Random Forests on GPUs☆705Updated 5 months ago
- Evaluation of the CNN design choices performance on ImageNet-2012.☆743Updated 8 years ago
- Python interface to GPU-powered libraries☆989Updated last year
- Open source implementation of "Spreading Vectors for Similarity Search"☆320Updated 4 years ago
- Navigating Spreading-out Graph For Approximate Nearest Neighbor Search☆695Updated last year
- Python framework for fast (approximated) nearest neighbour search in large, high-dimensional data sets using different locality-sensitive…☆769Updated 2 years ago
- Collective communications library with various primitives for multi-machine training.☆1,352Updated 3 weeks ago
- TVM integration into PyTorch☆453Updated 5 years ago
- Code for experiments regarding importance sampling for training neural networks☆329Updated 3 years ago
- Pure python implementation of product quantization for nearest neighbor search☆350Updated 3 months ago
- PyTorch layer-by-layer model profiler☆605Updated 4 years ago
- Training Very Deep Neural Networks Without Skip-Connections☆588Updated 7 years ago
- Metric learning algorithms in Python☆1,425Updated last year
- auto-tuning momentum SGD optimizer☆424Updated 7 years ago
- GPU Accelerated t-SNE for CUDA with Python bindings☆1,878Updated 11 months ago
- Approximate Nearest Neighbor Search for Sparse Data in Python!☆919Updated 4 years ago
- A CUDNN minimal deep learning training code sample using LeNet.☆268Updated 2 years ago
- Pytorch C++ Library☆367Updated 7 years ago
- CUDA Matrix Factorization Library with Alternating Least Square (ALS)☆180Updated 7 years ago