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☆298Updated 6 years ago
- Some CUDA design patterns and a bit of template magic for CUDA☆156Updated 2 years ago
- CUDA Data Parallel Primitives Library☆432Updated 6 years ago
- CUDA by practice☆129Updated 5 years ago
- CUDA official sample codes☆371Updated 9 years ago
- CNN accelerated by cuda. Test on mnist and finilly get 99.76%☆186Updated 7 years ago
- Example of how to use CUDA with CMake >= 3.8☆70Updated last month
- Efficient Top-K implementation on the GPU☆183Updated 6 years ago
- A CUDA implementation of the k-means clustering algorithm☆252Updated 13 years ago
- Source code that accompanies The CUDA Handbook.☆532Updated 5 months ago
- Python bindings for NVTX☆66Updated 2 years ago
- a c++/cuda template library for tensor lazy evaluation☆162Updated 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☆162Updated 5 months ago
- A minimalistic header only C++11 Neural Network library based on Eigen::Tensor☆20Updated 7 years ago
- CUDA FFT convolution☆15Updated 10 years ago
- A CUDNN minimal deep learning training code sample using LeNet.☆268Updated 2 years ago
- Guide for building custom op for TensorFlow☆382Updated 2 years ago
- TensorFlow and TVM integration☆37Updated 5 years ago
- ☆68Updated 11 years ago
- GPU-specialized parameter server for GPU machine learning.☆101Updated 7 years ago
- Example code to create and train a Pytorch model using the new C++ frontend.☆17Updated 6 years ago
- DLPack for Tensorflow☆35Updated 5 years ago
- Deep Learning/GPU Architect/Autonomous Driving Positions☆81Updated 5 years ago
- Explore the Capabilities of the TensorRT Platform☆264Updated 3 years ago
- Kernel Fusion and Runtime Compilation Based on NNVM☆70Updated 8 years ago
- Image classification C++ example using tensorflow shared library☆104Updated 8 years ago
- PyTorch C++ API Documentation☆232Updated last week
- Introduction to CUDA programming☆123Updated 8 years ago
- GPU-based large scale Approx. Nearest Neighbor Search, accepted at CVPR 2016☆92Updated 6 years ago