高性能短序列稀疏Mask Attention CUDA算子,针对<1K序列+75%稀疏度优化
☆79Mar 18, 2026Updated 4 months ago
Alternatives and similar repositories for sparse-mask-attention
Users that are interested in sparse-mask-attention are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Agentic Kernel Optimization for All — automated GPU kernel optimization for any kernel, any hardware, any language☆322May 31, 2026Updated last month
- LLM Inference via Triton (Flexible & Modular): Focused on Kernel Optimization using CUBIN binaries, Starting from gpt-oss Model☆119Apr 28, 2026Updated 2 months ago
- ☆37Aug 7, 2025Updated 11 months ago
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated 11 months ago
- 使用 cutlass 实现 flash-attention 精简版,具有教学意义☆59Aug 12, 2024Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Autoresearch for GPU kernels. Give it any PyTorch model, go to sleep, wake up to optimized Triton kernels.☆1,467Mar 19, 2026Updated 4 months ago
- cutile kernel examples☆50Apr 3, 2026Updated 3 months ago
- Some funny cute/cuteDSL code snippets☆33Mar 2, 2026Updated 4 months ago
- ☆46Updated this week
- High Performance LLM Inference Operator Library☆1,036Jul 2, 2026Updated 2 weeks ago
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- ☆229Updated this week
- ☆23Aug 20, 2025Updated 11 months ago
- Simple and efficient memory pool is implemented with C++11.☆10Jun 2, 2022Updated 4 years ago
- 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.
- Flash Attention in ~100 lines of CUDA (forward pass only)☆12Jun 10, 2024Updated 2 years ago
- A skill for automatically optimizing CUDA code.☆42Mar 26, 2026Updated 3 months ago
- ☆49Apr 15, 2024Updated 2 years ago
- GEMV implementation with CUTLASS☆21Aug 21, 2025Updated 10 months ago
- Mixed precision training from scratch with Tensors and CUDA☆30May 14, 2024Updated 2 years ago
- ☆66Feb 15, 2026Updated 5 months ago
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆157May 10, 2025Updated last year
- Build CUDA Neural Network From Scratch☆22Aug 28, 2024Updated last year
- 🎓Automatically Update circult-eda-mlsys-tinyml Papers Daily using Github Actions (Update Every 8th hours)☆10Jul 13, 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.
- ☆22Jun 10, 2026Updated last month
- Multiple GEMM operators are constructed with cutlass to support LLM inference.☆20Aug 3, 2025Updated 11 months ago
- 🤖FFPA: Extends FA-2/3 via Split-D for large headdims, 1.5x~6×↑🎉 vs SDPA, up to 513~535 TFLOPS🎉 on NVIDIA H200.☆315Updated this week
- ☆121May 16, 2025Updated last year
- ☆156Mar 4, 2025Updated last year
- ☆85Mar 31, 2026Updated 3 months ago
- 📚200+ Tensor/CUDA Cores Kernels, ⚡️flash-attn-mma, ⚡️hgemm with WMMA, MMA and CuTe (98%~100% TFLOPS of cuBLAS/FA2 🎉🎉).☆90Apr 26, 2025Updated last year
- ☆22Aug 14, 2024Updated last year
- 分享AI Infra知识&代码练习:PyTorch、vLLM/SGLang、slime/vime框架入门⚡️、性能加速🚀、大模型基础🧠、AI软硬件🔧等☆2,975Jul 2, 2026Updated 2 weeks ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- ☆24Sep 4, 2025Updated 10 months ago
- ☆13Dec 22, 2024Updated last year
- Awesome code, projects, books, etc. related to CUDA☆38Jun 2, 2026Updated last month
- A vector field rendering library☆17Jul 31, 2019Updated 6 years ago
- TensorRT 2022 亚军方案,tensorrt加速mobilevit模型☆70Jun 22, 2022Updated 4 years ago
- Triton implementation of FlashAttention2 that adds Custom Masks.☆176Aug 14, 2024Updated last year
- Autonomous GPU kernel optimization system driven by AI agents.☆31Mar 29, 2026Updated 3 months ago