flashinfer-ai / flashinfer
FlashInfer: Kernel Library for LLM Serving
See what the GitHub community is most excited about today.
FlashInfer: Kernel Library for LLM Serving
CUDA accelerated rasterization of gaussian splatting
NCCL Tests
LLM training in simple, raw C/CUDA
Causal depthwise conv1d in CUDA, with a PyTorch interface
DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling
DeepEP: an efficient expert-parallel communication library
Instant neural graphics primitives: lightning fast NeRF and more
Lightning fast differentiable SSIM.
Fast CUDA matrix multiplication from scratch
Tile primitives for speedy kernels
cuVS - a library for vector search and clustering on the GPU
GPU accelerated decision optimization
Quantized Attention achieves speedup of 2-5x and 3-11x compared to FlashAttention and xformers, without lossing end-to-end metrics across language, image, and video models.
CUDA Library Samples
CUDA Kernel Benchmarking Library
[ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl