gpu-mode / reference-kernelsLinks
Official Problem Sets / Reference Kernels for the GPU MODE Leaderboard!
☆200Updated last week
Alternatives and similar repositories for reference-kernels
Users that are interested in reference-kernels are comparing it to the libraries listed below
Sorting:
- ☆286Updated this week
- Helpful kernel tutorials and examples for tile-based GPU programming☆630Updated this week
- Fastest kernels written from scratch☆532Updated 4 months ago
- Cataloging released Triton kernels.☆291Updated 4 months ago
- ☆129Updated 3 months ago
- Fast low-bit matmul kernels in Triton☆427Updated this week
- A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS☆249Updated 9 months ago
- kernels, of the mega variety☆665Updated last week
- A Quirky Assortment of CuTe Kernels☆781Updated this week
- Learn CUDA with PyTorch☆193Updated this week
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆440Updated last month
- High-Performance FP32 GEMM on CUDA devices☆117Updated last year
- Step by step implementation of a fast softmax kernel in CUDA☆60Updated last year
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆324Updated this week
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆164Updated this week
- Evaluating Large Language Models for CUDA Code Generation ComputeEval is a framework designed to generate and evaluate CUDA code from Lar…☆96Updated 3 weeks ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆739Updated this week
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆461Updated last month
- Ship correct and fast LLM kernels to PyTorch☆140Updated 3 weeks ago
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆194Updated this week
- Efficient implementation of DeepSeek Ops (Blockwise FP8 GEMM, MoE, and MLA) for AMD Instinct MI300X☆75Updated this week
- ring-attention experiments☆165Updated last year
- Collection of kernels written in Triton language☆178Updated last week
- ☆102Updated last year
- Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools☆99Updated last week
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆155Updated 2 years ago
- Quantized LLM training in pure CUDA/C++.☆235Updated 2 weeks ago
- extensible collectives library in triton☆95Updated 10 months ago
- Applied AI experiments and examples for PyTorch☆315Updated 5 months ago
- This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".☆97Updated 4 months ago