Efficient and unified implementations for TopK-based sparse attention
☆41Jul 10, 2026Updated 3 months ago
Alternatives and similar repositories for flash-topk-attention
Users that are interested in flash-topk-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:
- Ulysses sequence-parallel all-to-all as a torch custom op, moved by the GPU copy engines into torch symmetric memory. Zero SM usage; 1.66…☆54Aug 25, 2026Updated last month
- ☆116Aug 5, 2026Updated 2 months ago
- NaiveRT: a self-contained TP8 inference runtime for the Naive N series on eight Blackwell GPUs with UMMA support. One persistent CUDA gra…☆30Sep 27, 2026Updated 3 weeks ago
- High performance RMSNorm Implement by using SM Core Storage(Registers and Shared Memory)☆31Jan 22, 2026Updated 8 months ago
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated 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.
- ☆32Jul 2, 2025Updated last year
- Reference implementation and examples of the CuTe Layout representation and algebra.☆379Sep 30, 2026Updated 2 weeks ago
- FLA but cuTile☆27Apr 17, 2026Updated 6 months ago
- ☆38Aug 4, 2026Updated 2 months ago
- DeepStack: Facilitating Co-Design Exploration of 3D DRAM-Stacked Accelerators for Distributed LLM Inference. Includes the MICRO 2026 AE a…☆65Oct 5, 2026Updated 2 weeks ago
- a size profiler for cuda binary☆74Oct 3, 2026Updated 2 weeks ago
- Automated CUDA kernel performance diagnostics from NVIDIA Nsight Compute (NCU) CSV exports.☆35Mar 18, 2026Updated 7 months ago
- Debug print operator for cudagraph debugging☆18Aug 2, 2024Updated 2 years ago
- ☆26Jun 10, 2026Updated 4 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.
- DeepSeek-V3.2-Exp DSA Warmup Lightning Indexer training operator based on tilelang☆52Nov 19, 2025Updated 11 months ago
- Muon in Int8 Precision Made Possible☆20Jun 18, 2026Updated 4 months ago
- ⚡️Qwen-Image 4.8x🎉 speedup with Hybrid Acceleration for low VRAM GPUs☆17Oct 24, 2025Updated 11 months ago
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- Curated high-performance TIRx GPU kernels☆32Updated this week
- ☆129Updated this week
- DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling☆46Updated this week
- ☆57Sep 27, 2026Updated 3 weeks ago
- ☆11Dec 26, 2025Updated 9 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- A toolkit for developers to simplify the transformation of nn.Module instances. It's now corresponding to Pytorch.fx.☆13Apr 7, 2023Updated 3 years ago
- The agent flow system for token maxxing.☆324Updated this week
- ☆54May 19, 2025Updated last year
- A high-throughput and memory-efficient inference and serving engine for LLMs☆17Jun 3, 2024Updated 2 years ago
- A linker for partial GPU kernels.☆27Oct 12, 2026Updated last week
- ☆51Aug 17, 2026Updated 2 months ago
- My love.☆28Apr 1, 2026Updated 6 months ago
- LLM Inference via Triton (Flexible & Modular): Focused on Kernel Optimization using CUBIN binaries, Starting from gpt-oss Model☆119Apr 28, 2026Updated 5 months ago
- An Optimizer for Nvidia Compilers.☆139Updated this week
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- DeeperGEMM: crazy optimized version☆85May 5, 2025Updated last year
- ☆58Apr 22, 2026Updated 5 months ago
- Skills for writing tilelang and debugging with CUDA toolkits.☆145Updated this week
- cutile kernel examples☆51Apr 3, 2026Updated 6 months ago
- A collection of megakernels for TPU☆164Updated this week
- ☆37Aug 7, 2025Updated last year
- NVFP4 Flash-Attention 4 on BlackWell☆68Sep 26, 2026Updated 3 weeks ago