All Projects → chrischoy → Pytorch_knn_cuda

chrischoy / Pytorch_knn_cuda

K-Nearest Neighbor in Pytorch

Labels

Projects that are alternatives of or similar to Pytorch knn cuda

Cuda Convnet2.torch
Torch7 bindings for cuda-convnet2 kernels!
Stars: ✭ 42 (-28.81%)
Mutual labels:  cuda
Carlsim3
CARLsim is an efficient, easy-to-use, GPU-accelerated software framework for simulating large-scale spiking neural network (SNN) models with a high degree of biological detail.
Stars: ✭ 52 (-11.86%)
Mutual labels:  cuda
Nvbio Gpl
NVBIO is a library of reusable components designed to accelerate bioinformatics applications using CUDA.
Stars: ✭ 56 (-5.08%)
Mutual labels:  cuda
Lyra
Stars: ✭ 43 (-27.12%)
Mutual labels:  cuda
Cs344
Introduction to Parallel Programming class code
Stars: ✭ 1,051 (+1681.36%)
Mutual labels:  cuda
Deformable conv2d pytorch
deformable_conv2d layer implemented in pytorch
Stars: ✭ 53 (-10.17%)
Mutual labels:  cuda
Sixtyfour
How fast can we brute force a 64-bit comparison?
Stars: ✭ 41 (-30.51%)
Mutual labels:  cuda
Cme213 material 2013
CME 213 Class Material
Stars: ✭ 57 (-3.39%)
Mutual labels:  cuda
Hungariangpu
An GPU/CUDA implementation of the Hungarian algorithm
Stars: ✭ 51 (-13.56%)
Mutual labels:  cuda
Hzproc
torch data augmentation toolbox (supports affine transform)
Stars: ✭ 56 (-5.08%)
Mutual labels:  cuda
Slic cuda
Superpixel SLIC for GPU (CUDA)
Stars: ✭ 45 (-23.73%)
Mutual labels:  cuda
Hornet
Hornet data structure for sparse dynamic graphs and matrices
Stars: ✭ 49 (-16.95%)
Mutual labels:  cuda
3d Ken Burns
an implementation of 3D Ken Burns Effect from a Single Image using PyTorch
Stars: ✭ 1,073 (+1718.64%)
Mutual labels:  cuda
Docs Pytorch
Deep Object Co-Segmentation
Stars: ✭ 43 (-27.12%)
Mutual labels:  cuda
Cuda Samples
Samples for CUDA Developers which demonstrates features in CUDA Toolkit
Stars: ✭ 1,087 (+1742.37%)
Mutual labels:  cuda
Qualia2.0
Qualia is a deep learning framework deeply integrated with automatic differentiation and dynamic graphing with CUDA acceleration. Qualia was built from scratch.
Stars: ✭ 41 (-30.51%)
Mutual labels:  cuda
Pamtri
PAMTRI: Pose-Aware Multi-Task Learning for Vehicle Re-Identification (ICCV 2019) - Official PyTorch Implementation
Stars: ✭ 53 (-10.17%)
Mutual labels:  cuda
Dokai
Collection of Docker images for ML/DL and video processing projects
Stars: ✭ 58 (-1.69%)
Mutual labels:  cuda
Heteroflow
Concurrent CPU-GPU Programming using Task Models
Stars: ✭ 57 (-3.39%)
Mutual labels:  cuda
Dink
点云深度学习框架 | Point cloud Deep learning Framework
Stars: ✭ 56 (-5.08%)
Mutual labels:  cuda

Pytorch KNN CUDA

  • 2019/11/02 This repository will no longer be maintained as pytorch supports sort() and kthvalue on tensors.
git clone https://github.com/chrischoy/pytorch_knn_cuda
cd pytorch_knn_cuda
make
python __init__.py
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].