Flash Attention from Scratch on CUDA Ampere
☆196Sep 1, 2025Updated last year
Alternatives and similar repositories for flash_attention_from_scratch
Users that are interested in flash_attention_from_scratch are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- From Minimal GEMM to Everything☆241Jul 9, 2026Updated last month
- Cute layout visualization☆45Jan 18, 2026Updated 7 months ago
- 使用 cutlass 实现 flash-attention 精简版，具有教学意义☆59Aug 12, 2024Updated 2 years ago
- ☆192May 11, 2026Updated 3 months ago
- Examples of CUDA implementations by Cutlass CuTe☆282Jul 1, 2025Updated last year
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- High Performance LLM Inference Operator Library☆1,146Updated this week
- GEMV implementation with CUTLASS☆21Aug 21, 2025Updated last year
- flash attention tutorial written in python, triton, cuda, cutlass☆534Jan 20, 2026Updated 7 months ago
- Kernel Library for Large Headdim Attention (64~1024, BF16/FP8/FP4), 1.5x~15x↑ vs PyTorch SDPA.☆331Updated this week
- ☆122May 16, 2025Updated last year
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆200Feb 11, 2026Updated 6 months ago
- LeetGPU Solutions☆128Oct 9, 2025Updated 10 months ago
- Modern CUDA Learn Notes with PyTorch for Beginners, 200+ CUDA Kernels, Tensor Cores, HGEMM, FA-2 MMA.☆11,900Updated this week
- A lightweight design for computation-communication overlap.☆246Jan 20, 2026Updated 7 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- CUTLASS and CuTe Examples☆140Nov 30, 2025Updated 9 months ago
- Implement Flash Attention using Cute.☆111Dec 17, 2024Updated last year
- Some funny cute/cuteDSL code snippets☆34Mar 2, 2026Updated 6 months ago
- A pure-Python implementation of the Nvidia CuTe layout algebra intended to be approachable and easy to learn.☆242Jun 29, 2026Updated 2 months ago
- NCU-driven iterative optimization workflow for CUDA/CUTLASS/Triton/CuTe DSL kernels.☆25Apr 10, 2026Updated 4 months ago
- ☆220Aug 26, 2026Updated last week
- Learn CUDA with PyTorch☆463Updated this week
- Fastest kernels written from scratch☆620Aug 15, 2026Updated 3 weeks ago
- ☆441Aug 26, 2026Updated last week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Puzzles for learning Triton, play it with minimal environment configuration!☆745Mar 17, 2026Updated 5 months ago
- ☆98Mar 26, 2025Updated last year
- Perplexity GPU Kernels☆607Nov 7, 2025Updated 10 months ago
- Low overhead tracing library and trace visualizer for pipelined CUDA kernels☆137Jul 29, 2026Updated last month
- A Quirky Assortment of CuTe Kernels☆1,137Updated this week
- ☆92Feb 5, 2026Updated 7 months ago
- ☆32Jul 2, 2025Updated last year
- The repository has collected a batch of noteworthy MLSys bloggers (Algorithms/Systems)☆345Jan 5, 2025Updated last year
- A compact implementation of SGLang, designed to demystify the complexities of modern LLM serving systems.☆4,990May 17, 2026Updated 3 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- RLVR training for LLM in CUDA/C++☆42Jul 25, 2026Updated last month
- FlashInfer: Kernel Library for LLM Serving☆6,343Updated this week
- A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS☆270May 6, 2025Updated last year
- Flash Attention in ~100 lines of CUDA (forward pass only)☆1,180Dec 30, 2024Updated last year
- 🍎 One kernel a day keeps high latency away. A hands-on CUDA learning path featuring a rich collection of kernels, from the basics to pea…☆214Aug 29, 2026Updated last week
- Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels☆7,359Updated this week
- ☆123May 31, 2025Updated last year