Implementation of FlashAttention (FA1-FA4) in PyTorch for educational and algorithmic clarity
☆228Apr 12, 2026Updated 6 months ago
Alternatives and similar repositories for FlashAttention-PyTorch
Users that are interested in FlashAttention-PyTorch are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernels☆116Jul 31, 2023Updated 3 years ago
- A simple but instructive implementation of DP, TP, FSDP, FSDP+TP using pytorch distributed primitives☆24Apr 12, 2026Updated 6 months ago
- flash attention tutorial written in python, triton, cuda, cutlass☆542Oct 22, 2026Updated 2 weeks ago
- a minimal cache manager for PagedAttention, on top of llama3.☆151Aug 26, 2024Updated 2 years ago
- Flash Attention in ~100 lines of CUDA (forward pass only)☆1,203Dec 30, 2024Updated last year
- 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.
- An approximate implementation of the OpenAI paper - An Empirical Model of Large-Batch Training for MNIST☆11Nov 19, 2022Updated 3 years ago
- ☆27Aug 5, 2022Updated 4 years ago
- Optimize SGEMM step by step☆10Sep 4, 2024Updated 2 years ago
- Triton implementation of FlashAttention2 that adds Custom Masks.☆180Aug 14, 2024Updated 2 years ago
- [ACM MM25] MCA-LLaVA: Manhattan Causal Attention for Reducing Hallucination in Large Vision-Language Models☆14Aug 8, 2025Updated last year
- Implement Flash Attention using Cute.☆113Dec 17, 2024Updated last year
- Prune transformer layers☆75Sep 23, 2026Updated last month
- ☆16Mar 13, 2023Updated 3 years ago
- Fast and memory-efficient exact attention☆23Oct 27, 2026Updated last week
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Fast and memory-efficient exact attention☆25,256Updated this week
- Experimental adapter for fine-tuning Qwen3-VL as a Vision-Language-Action (VLA) model☆15Dec 21, 2025Updated 10 months ago
- ☆165Sep 15, 2023Updated 3 years ago
- ☆40Dec 14, 2025Updated 10 months ago
- Implementation of the paper "Opcodes as predictor for malware " by Daniel Bilar☆11Oct 17, 2020Updated 6 years ago
- Seq2act: Mapping Natural Language Instructions to Mobile UI Action Sequences from Google research☆15Jul 13, 2020Updated 6 years ago
- Triton implement of bi-directional (non-causal) linear attention☆84Aug 10, 2026Updated 2 months ago
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆93Jul 17, 2025Updated last year
- [ACM MM 2025 Oral] TinyServe: Query-Aware Page Allocation Optimization☆11Oct 26, 2026Updated last week
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Sparse Attention with Linear Units☆20Apr 21, 2021Updated 5 years ago
- ☆156Mar 4, 2025Updated last year
- High Performance FP8 GEMM Kernels for SM89 and later GPUs.☆21Jan 24, 2025Updated last year
- ☆11Mar 29, 2025Updated last year
- Fast Hadamard transform in CUDA, with a PyTorch interface☆357Mar 10, 2026Updated 7 months ago
- Claude Code agent skills for writing optimized Triton GPU kernels☆15Mar 6, 2026Updated 8 months ago
- Flash-Flash KDA: H100-optimized Flash Kimi Delta Attention kernels☆24Jul 24, 2026Updated 3 months ago
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆155Feb 25, 2026Updated 8 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆113Sep 10, 2024Updated 2 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Learnings + Exercises from the PMPP book!☆11Mar 1, 2026Updated 8 months ago
- Puzzles for learning Triton☆2,663Apr 1, 2026Updated 7 months ago
- A CUDA kernel optimization toolkit for validation, benchmarking, Nsight Compute profiling, bottleneck analysis, and iterative tuning. It …☆217Sep 5, 2026Updated 2 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆787Aug 29, 2026Updated 2 months ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Jun 20, 2021Updated 5 years ago
- A simple high performance CUDA GEMM implementation.☆440Jan 4, 2024Updated 2 years ago
- Julia implementation of flash-attention operation for neural networks.☆11May 31, 2023Updated 3 years ago