LLM training in simple, raw C/CUDA

Instant neural graphics primitives: lightning fast NeRF and more

CUDA accelerated rasterization of gaussian splatting

Tile primitives for speedy kernels

Deformable ConvNets V2 (DCNv2) in PyTorch

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

[ICCV 2023] Official code for NeuS2

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

NeRFshop: Interactive Editing of Neural Radiance Fields

Neighborhood Attention Extension. Bringing attention to a neighborhood near you!

pytorch knn [cuda version]

Step-by-step optimization of CUDA SGEMM

A Neural Volumetric Renderer

Differentiable Iso-Surface Extraction Package (DISO)

Marching cubes implementation for PyTorch environment.

Fast Matrix Multiplications for Lookup Table-Quantized LLMs

A modular differential gaussian rasterization library.

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

PyTorch-Based Fast and Efficient Processing for Various Machine Learning Applications with Diverse Sparsity

Blazingly fast encoding for neural networks based on permutohedral lattices

Faster depthwise convolutions for PyTorch

Trilinear sampler with smoothstep and double backpropagation

Extension to `F.grid_sample` that allows using batch index per grid point.