chrischoy / Pytorch_knn_cuda
K-Nearest Neighbor in Pytorch
Stars: ✭ 59
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
Deformable conv2d pytorch
deformable_conv2d layer implemented in pytorch
Stars: ✭ 53 (-10.17%)
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
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
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
Pytorch KNN CUDA
- 2019/11/02 This repository will no longer be maintained as pytorch supports
sort()
andkthvalue
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].