mit-han-lab / nunchaku
[ICLR2025 Spotlight] SVDQuant: Absorbing Outliers by Low-Rank Components for 4-Bit Diffusion Models
See what the GitHub community is most excited about today.
[ICLR2025 Spotlight] SVDQuant: Absorbing Outliers by Low-Rank Components for 4-Bit Diffusion Models
LLM training in simple, raw C/CUDA
A repository where GPU applications are aggregated using a common build flow that supports multiple CUDA versions.
cuVS - a library for vector search and clustering on the GPU
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.
Quantized Attention that achieves speedups of 2.1-3.1x and 2.7-5.1x compared to FlashAttention2 and xformers, respectively, without lossing end-to-end metrics across various models.
Tile primitives for speedy kernels
FlashInfer: Kernel Library for LLM Serving
AI Tensor Engine for ROCm
CUDA Kernel Benchmarking Library
[ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl
Fast parallel CTC.
cuGraph - RAPIDS Graph Analytics Library
Causal depthwise conv1d in CUDA, with a PyTorch interface
Instant neural graphics primitives: lightning fast NeRF and more