Instant neural graphics primitives: lightning fast NeRF and more

Fast parallel CTC.

Deformable ConvNets V2 (DCNv2) in PyTorch

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

Reference implementation of Megalodon 7B model

Code for KiloNeRF: Speeding up Neural Radiance Fields with Thousands of Tiny MLPs

PopSift is an implementation of the SIFT algorithm in CUDA.

MegBA: A GPU-Based Distributed Library for Large-Scale Bundle Adjustment

NeRFshop: Interactive Editing of Neural Radiance Fields

GPU implementation of a fast generalized ANS (asymmetric numeral system) entropy encoder and decoder, with extensions for lossless compression of numerical and other data types in HPC/ML applications.

Chamfer Distance in Pytorch with f-score

Official implementation of "Decentralization and Acceleration Enables Large-Scale Bundle Adjustment"

Pytorch package to compute Chamfer distance between point sets (pointclouds).

[ECCV'2022 Oral] PyTorch implementation for: SimCC: a Simple Coordinate Classification Perspective for Human Pose Estimation (http://arxiv.org/abs/2107.03332). Old name: SimDR

PyTorch Wrapper for Earth-Mover-Distance (EMD) for 3D point cloud regression

Implementation of fused cosine similarity attention in the same style as Flash Attention

Planar Prior Assisted PatchMatch Multi-View Stereo

Fixed Radius Nearest Neighbor Search on GPU

GGNN: State of the Art Graph-based GPU Nearest Neighbor Search

Multi-Scale Geometric Consistency Guided Multi-View Stereo

A CUDA Mesh RayTracer with BVH acceleration, with python bindings and a GUI.

A CUDA Extension of Neural Network Libraries

Instant NeRF for depth/3d shape estimation of transparent objects

deformable_conv2d layer implemented in pytorch

Test-Time Personalization with a Transformer for Human Pose Estimation, NeurIPS 2021

Adapted signed distance function (SDF) for detecting collisions between 3D interacting hands.

ray cast mesh to get normal, depth and face_ids