src-d / kmcudaLinks
Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA
☆833Updated 3 years ago
Alternatives and similar repositories for kmcuda
Users that are interested in kmcuda are comparing it to the libraries listed below
Sorting:
- A CUDA implementation of the k-means clustering algorithm☆252Updated 13 years ago
- ATen: A TENsor library for C++11☆703Updated 5 years ago
- fast library for ANN search and KNN graph construction☆296Updated 7 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,061Updated 2 years ago
- Fast k nearest neighbor search using GPU☆542Updated 7 years ago
- FAst Lookups of Cosine and Other Nearest Neighbors (based on fast locality-sensitive hashing)☆1,154Updated last year
- A library for k-nearest neighbor search☆385Updated last year
- Tutorial for building a custom CUDA function for Pytorch☆517Updated 6 years ago
- Python framework for fast (approximated) nearest neighbour search in large, high-dimensional data sets using different locality-sensitive…☆770Updated 2 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
- Python interface to GPU-powered libraries☆988Updated last year
- ☆714Updated 2 years ago
- Make huge neural nets fit in memory☆2,814Updated 5 years ago
- A lightweight library for PyTorch training tools and utilities☆1,713Updated this week
- Collective communications library with various primitives for multi-machine training.☆1,357Updated last month
- Barnes-Hut t-SNE☆910Updated 4 years ago
- Fast Recurrent Networks Library☆577Updated 9 years ago
- ThunderSVM: A Fast SVM Library on GPUs and CPUs☆1,614Updated last year
- Training Very Deep Neural Networks Without Skip-Connections☆588Updated 7 years ago
- ☆598Updated 7 years ago
- Python module for performing basic dense linear algebra computations on the GPU using CUDA.☆609Updated 5 years ago
- ThunderGBM: Fast GBDTs and Random Forests on GPUs☆706Updated 6 months ago
- Cython implementation of k-MC2 and AFK-MC2 seeding☆205Updated 4 years ago
- Matrix Shadow:Lightweight CPU/GPU Matrix and Tensor Template Library in C++/CUDA for (Deep) Machine Learning☆1,117Updated 6 years ago
- auto-tuning momentum SGD optimizer☆424Updated 7 years ago
- C++ extensions in PyTorch☆1,148Updated 3 months ago
- Memory consumption and FLOP count estimates for convnets☆928Updated 6 years ago
- GPU Accelerated t-SNE for CUDA with Python bindings☆1,882Updated last year
- Code repository for the paper "Hyperparameter Optimization: A Spectral Approach" by Elad Hazan, Adam Klivans, Yang Yuan.☆174Updated 6 years ago
- Acceleration package for neural networks on multi-core CPUs☆1,699Updated last year