src-d / kmcudaLinks
Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA
☆834Updated 2 years ago
Alternatives and similar repositories for kmcuda
Users that are interested in kmcuda are comparing it to the libraries listed below
Sorting:
- fast library for ANN search and KNN graph construction☆294Updated 7 years ago
- A CUDA implementation of the k-means clustering algorithm☆252Updated 13 years ago
- ATen: A TENsor library for C++11☆703Updated 5 years ago
- Training of Locally Optimized Product Quantization (LOPQ) models for approximate nearest neighbor search of high dimensional data in Pyth…☆568Updated 6 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,043Updated 2 years ago
- A library for k-nearest neighbor search☆384Updated last year
- ThunderSVM: A Fast SVM Library on GPUs and CPUs☆1,608Updated last year
- ☆716Updated 2 years ago
- FAst Lookups of Cosine and Other Nearest Neighbors (based on fast locality-sensitive hashing)☆1,153Updated last year
- Tutorial for building a custom CUDA function for Pytorch☆519Updated 6 years ago
- C++ extensions in PyTorch☆1,126Updated last month
- Python framework for fast (approximated) nearest neighbour search in large, high-dimensional data sets using different locality-sensitive…☆767Updated 2 years ago
- Make huge neural nets fit in memory☆2,805Updated 5 years ago
- A lightweight library for PyTorch training tools and utilities☆1,701Updated this week
- Simple package that makes your generator work in background thread☆280Updated 3 years ago
- Barnes-Hut t-SNE☆902Updated 4 years ago
- A language-agnostic interface to TensorBoard☆779Updated 7 years ago
- Evaluation of the CNN design choices performance on ImageNet-2012.☆743Updated 8 years ago
- Pure python implementation of product quantization for nearest neighbor search☆353Updated 2 months ago
- ThunderGBM: Fast GBDTs and Random Forests on GPUs☆704Updated 4 months ago
- Python interface to GPU-powered libraries☆994Updated last year
- A c++ toolbox of locality-sensitive hashing (LSH), provides several popular LSH algorithms, also support python and matlab.☆288Updated 8 years ago
- Standalone TensorBoard for visualizing in deep learning☆369Updated 5 years ago
- Spatially-sparse convolutional networks. Allows processing of sparse 2, 3 and 4 dimensional data.Build CNNs on the square/cubic/hypercubi…☆408Updated 4 years ago
- A fast Python implementation of locality sensitive hashing.☆667Updated 5 years ago
- Profiling and inspecting memory in pytorch☆1,065Updated last year
- PyTorch and Tensorflow functional model definitions☆587Updated 7 years ago
- Log TensorBoard events without touching TensorFlow☆628Updated 2 years ago
- Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets☆308Updated 8 years ago
- Deep Learning Benchmark for comparing the performance of DL frameworks, GPUs, and single vs half precision☆430Updated 5 years ago