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☆299Updated 6 years ago
- CUDA Data Parallel Primitives Library☆436Updated 7 years ago
- Some CUDA design patterns and a bit of template magic for CUDA☆156Updated 2 years ago
- Efficient Top-K implementation on the GPU☆187Updated 6 years ago
- Source code that accompanies The CUDA Handbook.☆550Updated last month
- CUDA official sample codes☆370Updated 10 years ago
- Python bindings for NVTX☆66Updated 2 years ago
- CNN accelerated by cuda. Test on mnist and finilly get 99.76%☆185Updated 8 years ago
- flexible-gemm conv of deepcore☆17Updated 5 years ago
- a c++/cuda template library for tensor lazy evaluation☆163Updated 2 years ago
- Kernel Fusion and Runtime Compilation Based on NNVM☆71Updated 8 years ago
- THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.☆85Updated last year
- Example of how to use CUDA with CMake >= 3.8☆70Updated 5 months ago
- RDMA Optimization on MXNet☆14Updated 8 years ago
- A CUDA implementation of the k-means clustering algorithm☆253Updated 13 years ago
- TensorFlow and TVM integration☆36Updated 5 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 6 years ago
- GGNN: State of the Art Graph-based GPU Nearest Neighbor Search☆166Updated 9 months ago
- CUDA by practice☆130Updated 5 years ago
- Deep Learning/GPU Architect/Autonomous Driving Positions☆81Updated 5 years ago
- A CUDNN minimal deep learning training code sample using LeNet.☆269Updated 2 years ago
- GPU-specialized parameter server for GPU machine learning.☆102Updated 7 years ago
- Optimized half precision gemm assembly kernels (deprecated due to ROCm)☆47Updated 8 years ago
- PyTorch C++ API Documentation☆240Updated this week
- A Tool for Automatic Parallelization of Deep Learning Training in Distributed Multi-GPU Environments.☆132Updated 3 years ago
- ☆44Updated 7 years ago
- ☆71Updated 11 years ago
- CLTune: An automatic OpenCL & CUDA kernel tuner☆182Updated 2 years ago
- Tests and benchmarks for cudnn (and in the future, other nvidia libraries)☆53Updated 4 years ago
- ☆126Updated 7 years ago