NVIDIA / kmeans
kmeans clustering with multi-GPU capabilities
☆120Updated last year
Alternatives and similar repositories for kmeans:
Users that are interested in kmeans are comparing it to the libraries listed below
- Python bindings for NVTX☆66Updated last year
- Efficient Top-K implementation on the GPU☆168Updated 5 years ago
- A simple memory manager for CUDA designed to help Deep Learning frameworks manage memory☆297Updated 6 years ago
- Example of how to use CUDA with CMake >= 3.8☆69Updated last year
- Kernel Fusion and Runtime Compilation Based on NNVM☆70Updated 8 years ago
- kmeans☆54Updated 8 years ago
- THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.☆84Updated last year
- ☆21Updated 7 years ago
- This repository contains the results and code for the MLPerf™ Training v0.5 benchmark.☆35Updated last year
- cuDNN sample codes provided by Nvidia☆45Updated 6 years ago
- CUDA official sample codes☆365Updated 9 years ago
- Tools and extensions for CUDA profiling☆65Updated 5 years ago
- ☆127Updated 7 years ago
- Some CUDA design patterns and a bit of template magic for CUDA☆149Updated last year
- flexible-gemm conv of deepcore☆17Updated 5 years ago
- Codebase associated with the PyTorch compiler tutorial☆46Updated 5 years ago
- Sparse matrix computation library for GPU☆54Updated 4 years ago
- Deep Learning/GPU Architect/Autonomous Driving Positions☆80Updated 5 years ago
- oneCCL Bindings for Pytorch*☆90Updated last week
- A CUDNN minimal deep learning training code sample using LeNet.☆264Updated last year
- This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.☆56Updated last year
- Optimized half precision gemm assembly kernels (deprecated due to ROCm)☆47Updated 7 years ago
- this is the release repository of superneurons☆52Updated 4 years ago
- Multi-GPU Computing Benchmark Suite (CUDA)☆42Updated 7 years ago
- GPU-specialized parameter server for GPU machine learning.☆100Updated 6 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 5 years ago
- Subpart source code of of deepcore v0.7☆27Updated 4 years ago
- CUDA Tensor Transpose (cuTT) library☆51Updated 7 years ago
- A warp-oriented dynamic hash table for GPUs☆73Updated last year
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆64Updated 3 years ago