NVIDIA / kmeansLinks
kmeans clustering with multi-GPU capabilities
☆119Updated 2 years ago
Alternatives and similar repositories for kmeans
Users that are interested in kmeans are comparing it to the libraries listed below
Sorting:
- A simple memory manager for CUDA designed to help Deep Learning frameworks manage memory☆296Updated 6 years ago
- Python bindings for NVTX☆66Updated 2 years ago
- Efficient Top-K implementation on the GPU☆179Updated 6 years ago
- Some CUDA design patterns and a bit of template magic for CUDA☆154Updated 2 years ago
- THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.☆84Updated last year
- GGNN: State of the Art Graph-based GPU Nearest Neighbor Search☆161Updated 4 months ago
- GPU-based large scale Approx. Nearest Neighbor Search, accepted at CVPR 2016☆92Updated 6 years ago
- Kernel Fusion and Runtime Compilation Based on NNVM☆70Updated 8 years ago
- flexible-gemm conv of deepcore☆17Updated 5 years ago
- A warp-oriented dynamic hash table for GPUs☆73Updated last year
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 6 years ago
- Code for testing the native float16 matrix multiplication performance on Tesla P100 and V100 GPU based on cublasHgemm☆34Updated 5 years ago
- PQ Fast Scan☆62Updated 6 years ago
- This repository contains the results and code for the MLPerf™ Training v1.0 benchmark.☆38Updated last year
- GPU-specialized parameter server for GPU machine learning.☆101Updated 7 years ago
- BGHT: High-performance static GPU hash tables.☆66Updated 2 months ago
- Example repository for custom C++/CUDA operators for TorchScript☆114Updated 2 years ago
- A CUDNN minimal deep learning training code sample using LeNet.☆267Updated last year
- CUDA Data Parallel Primitives Library☆431Updated 6 years ago
- Codebase associated with the PyTorch compiler tutorial☆46Updated 5 years ago
- This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.☆56Updated 2 years ago
- Guide for building custom op for TensorFlow☆382Updated 2 years ago
- portDNN is a library implementing neural network algorithms written using SYCL☆113Updated last year
- this is the release repository of superneurons☆52Updated 4 years ago
- This repository contains the results and code for the MLPerf™ Training v0.5 benchmark.☆35Updated last month
- ☆125Updated 7 years ago
- ☆371Updated 7 years ago
- A tensor-aware point-to-point communication primitive for machine learning☆258Updated 2 years ago
- ☆91Updated 8 years ago
- A library of GPU kernels for sparse matrix operations.☆265Updated 4 years ago