Repository Index

Instant neural graphics primitives: lightning fast NeRF and more

• neural-network
• machine-learning
• cuda
• nerf

HVM1 (2022): a massively parallel, optimal functional runtime in Rust

DeepEP: an efficient expert-parallel communication library

FlashInfer: Kernel Library for LLM Serving

• gpu
• large-large-models
• cuda
• pytorch

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"

• human-pose-estimation
• deep-learning
• coco-keypoints-detection
• mpii-dataset

[ICLR2025, ICML2025, NeurIPS2025 Spotlight] Quantized Attention achieves speedup of 2-5x compared to FlashAttention, without losing end-to-end metrics across language, image, and video models.

• attention
• inference-acceleration
• llm
• quantization

Squeeze-and-Excitation Networks

• senet
• caffe
• gpu

Sample codes for my CUDA programming book

• cuda-programming
• gpu-programming
• molecular-dynamics-simulation

GPU Accelerated t-SNE for CUDA with Python bindings

• cuda
• gpu
• mnist
• tsne

A GPU implementation of Convolutional Neural Nets in C++