☆239Jan 2, 2025Updated last year
Alternatives and similar repositories for triton-flash-attention
Users that are interested in triton-flash-attention are comparing it to the libraries listed below
Sorting:
- ☆15Feb 23, 2025Updated last year
- Distributed training (multi-node) of a Transformer model☆94Apr 10, 2024Updated last year
- ML algorithms implementations that are good for learning the underlying principles☆27Dec 7, 2024Updated last year
- Recreating the minimal training methods of DeepSeek-R1 for small langauge models.☆22Feb 10, 2025Updated last year
- Material for gpu-mode lectures☆5,800Feb 1, 2026Updated last month
- flash attention tutorial written in python, triton, cuda, cutlass☆490Jan 20, 2026Updated last month
- making the official triton tutorials actually comprehensible☆131Aug 25, 2025Updated 6 months ago
- Puzzles for learning Triton☆2,324Nov 18, 2024Updated last year
- Stable Diffusion implemented from scratch in PyTorch☆1,030Oct 22, 2024Updated last year
- Coding a Multimodal (Vision) Language Model from scratch in PyTorch with full explanation: https://www.youtube.com/watch?v=vAmKB7iPkWw☆593Dec 6, 2024Updated last year
- 使用 cutlass 实现 flash-attention 精简版，具有教学意义☆58Aug 12, 2024Updated last year
- 100 days of building GPU kernels!☆575Apr 27, 2025Updated 10 months ago
- Optimize GEMM with tensorcore step by step☆36Dec 17, 2023Updated 2 years ago
- Transformers components but in Triton☆34May 9, 2025Updated 10 months ago
- ☆46May 24, 2025Updated 9 months ago
- Flash Attention in ~100 lines of CUDA (forward pass only)☆1,084Dec 30, 2024Updated last year
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,099Aug 26, 2025Updated 6 months ago
- Efficient Triton Kernels for LLM Training☆6,189Updated this week
- LLaMA 2 implemented from scratch in PyTorch☆365Sep 25, 2023Updated 2 years ago
- This project is about convolution operator optimization on GPU, include GEMM based (Implicit GEMM) convolution.☆43Sep 29, 2025Updated 5 months ago
- Implement Flash Attention using Cute.☆102Dec 17, 2024Updated last year
- Notes on Direct Preference Optimization☆24Apr 14, 2024Updated last year
- ☆10Oct 9, 2025Updated 5 months ago
- Linear Relational Embeddings (LREs) and Linear Relational Concepts (LRCs) for LLMs in PyTorch☆10Aug 7, 2024Updated last year
- 𝐈𝐬𝐥𝐚𝐦𝐢𝐜𝐓𝐫𝐚𝐧𝐬𝐥𝐚𝐭𝐨𝐫 is an automated solution designed to translate 𝐇𝐚𝐝𝐢𝐭𝐡𝐬 into multiple languages using the power …☆11Jan 17, 2025Updated last year
- Penn CIS 5650 (GPU Programming and Architecture) Final Project☆44Dec 11, 2023Updated 2 years ago
- 🚀 Efficient implementations of state-of-the-art linear attention models☆4,474Updated this week
- ☆116May 16, 2025Updated 9 months ago
- GPU Kernels☆221Apr 27, 2025Updated 10 months ago
- GPU programming related news and material links☆2,010Sep 17, 2025Updated 5 months ago
- Delay Differential Equations in Haskell☆11Dec 4, 2018Updated 7 years ago
- ☆11Dec 22, 2024Updated last year
- CUDA Matrix Multiplication Optimization☆261Jul 19, 2024Updated last year
- A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.☆464Mar 10, 2025Updated 11 months ago
- From scratch implementation of a vision language model in pure PyTorch☆253May 6, 2024Updated last year
- Tile primitives for speedy kernels☆3,202Feb 24, 2026Updated last week
- aesthetic tensor visualiser☆27Apr 23, 2025Updated 10 months ago
- A light llama-like llm inference framework based on the triton kernel.☆172Jan 5, 2026Updated 2 months ago
- FlagGems is an operator library for large language models implemented in the Triton Language.☆909Updated this week