NVIDIA / CUDALibrarySamples
CUDA Library Samples
See what the GitHub community is most excited about this month.
CUDA Library Samples
NCCL Tests
A massively parallel, optimal functional runtime in Rust
CUDA accelerated rasterization of gaussian splatting
An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
FlashInfer: Kernel Library for LLM Serving
Graphics Processing Units Molecular Dynamics
[MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.
🎉CUDA 笔记 / 大模型手撕CUDA / C++笔记,更新随缘: flash_attn、sgemm、sgemv、warp reduce、block reduce、dot product、elementwise、softmax、layernorm、rmsnorm、hist etc.
CUDA checkpoint and restore utility
how to optimize some algorithm in cuda.
A Real-Time Depth Sensor Simulator with GPU Acceleration
Reference implementation of Megalodon 7B model
Distributed multigrid linear solver library on GPU
Causal depthwise conv1d in CUDA, with a PyTorch interface
Sample codes for my CUDA programming book
cuVS - a library for vector search and clustering on the GPU
cuGraph - RAPIDS Graph Analytics Library
Training materials associated with NVIDIA's CUDA Training Series (www.olcf.ornl.gov/cuda-training-series/)
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.
CUDA Kernel Benchmarking Library