Code and data for paper "Deep Painterly Harmonization": https://arxiv.org/abs/1804.03189

The project is an official implementation of our CVPR2019 paper "Deep High-Resolution Representation Learning for Human Pose Estimation"

Squeeze-and-Excitation Networks

cuGraph - RAPIDS Graph Analytics Library

[MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.

Pytorch Bindings for warp-ctc

Fast k nearest neighbor search using GPU

Training materials associated with NVIDIA's CUDA Training Series (www.olcf.ornl.gov/cuda-training-series/)

Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda

This is a Tensor Train based compression library to compress sparse embedding tables used in large-scale machine learning models such as recommendation and natural language processing. We showed th…

Fixed Radius Nearest Neighbor Search on GPU