MetaX-MACA / FlashMLALinks
Fast and efficient attention method exploration and implementation.
☆21Updated 3 months ago
Alternatives and similar repositories for FlashMLA
Users that are interested in FlashMLA are comparing it to the libraries listed below
Sorting:
- An unofficial cuda assembler, for all generations of SASS, hopefully :)☆83Updated 2 years ago
- AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…☆251Updated 3 weeks ago
- A prefill & decode disaggregated LLM serving framework with shared GPU memory and fine-grained compute isolation.☆102Updated 2 months ago
- ☆80Updated this week
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆87Updated 2 months ago
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆184Updated 5 months ago
- This is a demo how to write a high performance convolution run on apple silicon☆54Updated 3 years ago
- A Python script to convert the output of NVIDIA Nsight Systems (in SQLite format) to JSON in Google Chrome Trace Event Format.☆38Updated 5 months ago
- ☆58Updated 4 years ago
- Fast and memory-efficient exact attention☆81Updated last week
- Ahead of Time (AOT) Triton Math Library☆70Updated last week
- Triton adapter for Ascend. Mirror of https://gitee.com/ascend/triton-ascend☆59Updated this week
- Automated machine learning as an AI-HPC benchmark☆66Updated 3 years ago
- heterogeneity-aware-lowering-and-optimization☆255Updated last year
- CVFusion is an open-source deep learning compiler to fuse the OpenCV operators.☆30Updated 2 years ago
- A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores☆52Updated last year
- Assembler and Decompiler for NVIDIA (Maxwell Pascal Volta Turing Ampere) GPUs.☆81Updated 2 years ago
- ☆14Updated 3 years ago
- An extension library of WMMA API (Tensor Core API)☆99Updated last year
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆93Updated this week
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆79Updated 8 months ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆121Updated 3 years ago
- Multi-GPU communication profiler and visualizer☆31Updated last year
- Tacker: Tensor-CUDA Core Kernel Fusion for Improving the GPU Utilization while Ensuring QoS☆29Updated 5 months ago
- FlagTree is a unified compiler for multiple AI chips, which is forked from triton-lang/triton.☆64Updated this week
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆91Updated 3 weeks ago
- ☆40Updated this week
- Standalone Flash Attention v2 kernel without libtorch dependency☆111Updated 10 months ago
- GVProf: A Value Profiler for GPU-based Clusters☆51Updated last year
- A novel temporal fusion framework for propelling autoregressive model inference☆11Updated this week