karpathy / llm.c
LLM training in simple, raw C/CUDA
See what the GitHub community is most excited about today.
LLM training in simple, raw C/CUDA
CUDA accelerated rasterization of gaussian splatting
Causal depthwise conv1d in CUDA, with a PyTorch interface
FlashInfer: Kernel Library for LLM Serving
NCCL Tests
Instant neural graphics primitives: lightning fast NeRF and more
CUDA Library Samples
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.
Tile primitives for speedy kernels
[MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.
cuGraph - RAPIDS Graph Analytics Library
[ARCHIVED] Cooperative primitives for CUDA C++. See https://github.com/NVIDIA/cccl