hkproj / triton-flash-attentionView external linksLinks
☆237Jan 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:
- 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
- Material for gpu-mode lectures☆5,726Feb 1, 2026Updated 2 weeks ago
- making the official triton tutorials actually comprehensible☆115Aug 25, 2025Updated 5 months ago
- Puzzles for learning Triton☆2,296Nov 18, 2024Updated last year
- Stable Diffusion implemented from scratch in PyTorch☆1,030Oct 22, 2024Updated last year
- flash attention tutorial written in python, triton, cuda, cutlass☆486Jan 20, 2026Updated 3 weeks ago
- Coding a Multimodal (Vision) Language Model from scratch in PyTorch with full explanation: https://www.youtube.com/watch?v=vAmKB7iPkWw☆594Dec 6, 2024Updated last year
- 使用 cutlass 实现 flash-attention 精简版，具有教学意义☆56Aug 12, 2024Updated last year
- 100 days of building GPU kernels!☆570Apr 27, 2025Updated 9 months ago
- Optimize GEMM with tensorcore step by step☆36Dec 17, 2023Updated 2 years ago
- Transformers components but in Triton☆34May 9, 2025Updated 9 months ago
- Code repository for ICLR 2025 paper "LeanQuant: Accurate and Scalable Large Language Model Quantization with Loss-error-aware Grid"☆24Mar 2, 2025Updated 11 months ago
- A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS☆252May 6, 2025Updated 9 months ago
- ☆46May 24, 2025Updated 8 months ago
- Flash Attention in ~100 lines of CUDA (forward pass only)☆1,068Dec 30, 2024Updated last year
- Notes on quantization in neural networks☆119Dec 14, 2023Updated 2 years ago
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,076Aug 26, 2025Updated 5 months ago
- Efficient Triton Kernels for LLM Training☆6,141Updated this week
- LLaMA 2 implemented from scratch in PyTorch☆366Sep 25, 2023Updated 2 years ago
- This project is about convolution operator optimization on GPU, include GEMM based (Implicit GEMM) convolution.☆43Sep 29, 2025Updated 4 months ago
- Implement Flash Attention using Cute.☆100Dec 17, 2024Updated last year
- Notes on Direct Preference Optimization☆24Apr 14, 2024Updated last year
- Penn CIS 5650 (GPU Programming and Architecture) Final Project☆44Dec 11, 2023Updated 2 years ago
- ☆10Oct 9, 2025Updated 4 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
- GPU Kernels☆220Apr 27, 2025Updated 9 months ago
- From Minimal GEMM to Everything☆139Feb 10, 2026Updated last week
- ☆114May 16, 2025Updated 9 months ago
- GPU programming related news and material links☆1,967Sep 17, 2025Updated 4 months ago
- 🚀 Efficient implementations of state-of-the-art linear attention models☆4,379Updated this week
- ☆11Dec 22, 2024Updated last year
- CUDA Matrix Multiplication Optimization☆257Jul 19, 2024Updated last year
- A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.☆457Mar 10, 2025Updated 11 months ago
- Tile primitives for speedy kernels☆3,139Updated this week
- A light llama-like llm inference framework based on the triton kernel.☆171Jan 5, 2026Updated last month
- FlagGems is an operator library for large language models implemented in the Triton Language.☆898Updated this week
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 2 years ago