Dao-AILab / causal-conv1d
Causal depthwise conv1d in CUDA, with a PyTorch interface
See what the GitHub community is most excited about this month.
Causal depthwise conv1d in CUDA, with a PyTorch interface
FlashInfer: Kernel Library for LLM Serving
Tile primitives for speedy kernels
cuGraph - RAPIDS Graph Analytics Library
Instant neural graphics primitives: lightning fast NeRF and more
NCCL Tests
CUDA Library Samples
CUDA accelerated rasterization of gaussian splatting
CUDA Kernel Benchmarking Library
Distributed multigrid linear solver library on GPU
[MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.
A massively parallel, optimal functional runtime in Rust
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.