NVIDIA / cub
[ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
See what the GitHub community is most excited about today.
[ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
Instant neural graphics primitives: lightning fast NeRF and more
Tile primitives for speedy kernels
FlashInfer: Kernel Library for LLM Serving
RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing high performance applications.
A massively parallel, optimal functional runtime in Rust
cuVS - a library for vector search and clustering on the GPU
cuGraph - RAPIDS Graph Analytics Library
Causal depthwise conv1d in CUDA, with a PyTorch interface
CUDA Kernel Benchmarking Library
LLM training in simple, raw C/CUDA
The project is an official implementation of our CVPR2019 paper "Deep High-Resolution Representation Learning for Human Pose Estimation"
Neighborhood Attention Extension. Bringing attention to a neighborhood near you!