kmcuda

GPU-accelerated clustering library

Optimized K-means and K-nn implementation using NVIDIA CUDA with low memory consumption

Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA

GitHub

809 stars
28 watching
146 forks
Language: Jupyter Notebook
last commit: over 2 years ago
Linked from 1 awesome list

afk-mc2cudahacktoberfestkmeansknn-searchmachine-learningpythonyinyang

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
xtra-computing/thundergbm Accelerates machine learning algorithms on GPUs to improve performance and efficiency 695
src-d/minhashcuda An implementation of Weighted MinHash calculation on CUDA for efficient similarity estimation and compression of large data sets. 114
kalvar/ios-krkmeans-algorithm K-Means clustering algorithm implementation with multi-dimensional support and customizable features 23
uncomplicate/bayadera Enables Bayesian data analysis and machine learning on graphics processing units (GPUs) to accelerate computational tasks. 365
nvidia/grcuda Enables efficient data exchange and invocation of existing GPU kernels from host languages in the GraalVM 223
uncomplicate/clojurecuda Provides Clojure bindings for CUDA programming, enabling GPU-accelerated computation in Clojure applications. 181
hkust-knowcomp/r-net An implementation of R-NET, a machine reading comprehension model using scaled multiplicative attention and variational dropout. 578
srvm/cupti_profiler A CUDA profiling library that allows developers to collect metrics and events from within CUDA applications. 37
reddavis/knn An implementation of the K Nearest Neighbour algorithm in Ruby, allowing users to query a dataset and retrieve the k most similar data points based on various distance measurements. 37
nvidia/matx A C++17 GPU-accelerated numerical computing library with Python-like syntax 1,229
jlblancoc/nanoflann A C++ library for efficient nearest neighbor search with KD-trees 2,269
neilkuan/rke2-eks-d-lab A sample project demonstrating deployment of a Kubernetes cluster using RKE2 and Amazon EKS-D with CDK 0
torch/cunn A CUDA implementation of neural network modules and related GPU operations 215
rust-math/intel-mkl-src A crate that redistributes and provides Rust bindings to Intel's Math Kernel Library (MKL) for scientific computations. 82
coreylowman/dfdx A deep learning library for Rust with GPU acceleration and ergonomic API. 1,754