Alternatives and similar repositories for FlashMLA

Users that are interested in FlashMLA are comparing it to the libraries listed below

• FlagTree / flagtree
  FlagTree is a unified compiler for multiple AI chips, which is forked from triton-lang/triton.
  ☆76Updated last week
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆109Updated 3 months ago
• OpenPPL / CuAssembler
  An unofficial cuda assembler, for all generations of SASS, hopefully ：）
  ☆84Updated 2 years ago
• bytedance / ByteMLPerf
  AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…
  ☆261Updated 2 weeks ago
• infinigence / Semi-PD
  A prefill & decode disaggregated LLM serving framework with shared GPU memory and fine-grained compute isolation.
  ☆105Updated 3 months ago
• pigirons / conv3x3_m1
  This is a demo how to write a high performance convolution run on apple silicon
  ☆54Updated 3 years ago
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆76Updated last week
• TiledTensor / TiledCUDA
  We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …
  ☆183Updated 7 months ago
• vllm-project / flash-attention
  Fast and memory-efficient exact attention
  ☆91Updated last week
• FlagOpen / FlagCX
  ☆85Updated this week
• QianyanTech / NBAssembler
  Assembler and Decompiler for NVIDIA (Maxwell Pascal Volta Turing Ampere) GPUs.
  ☆84Updated 2 years ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆110Updated 11 months ago
• Ascend / triton-ascend
  Triton adapter for Ascend. Mirror of https://gitee.com/ascend/triton-ascend
  ☆68Updated this week
• chenyu-jiang / nsys2json
  A Python script to convert the output of NVIDIA Nsight Systems (in SQLite format) to JSON in Google Chrome Trace Event Format.
  ☆39Updated last month
• ROCm / TransformerEngine
  ☆42Updated last week
• infinigence / FlashOverlap
  A lightweight design for computation-communication overlap.
  ☆161Updated this week
• PanZaifeng / RecFlex
  A recommendation model kernel optimizing system
  ☆10Updated 3 months ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆103Updated last year
• Ascend / torchair
  ☆18Updated this week
• tlc-pack / cutlass_fpA_intB_gemm
  A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer
  ☆94Updated this week
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆95Updated 2 months ago
• BBuf / tensorrt-llm-moe
  ☆31Updated 7 months ago
• apuaaChen / EVT_AE
  Artifacts of EVT ASPLOS'24
  ☆26Updated last year
• OpenPPL / ppl.llm.kernel.cuda
  ☆150Updated 7 months ago
• UDC-GAC / venom
  A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores
  ☆53Updated last year
• Bruce-Lee-LY / cuda_hgemv
  Several optimization methods of half-precision general matrix vector multiplication (HGEMV) using CUDA core.
  ☆64Updated 11 months ago
• muriloboratto / NCCL
  Sample examples of how to call collective operation functions on multi-GPU environments. A simple example of using broadcast, reduce, all…
  ☆34Updated 2 years ago
• HPMLL / NVIDIA-Hopper-Benchmark
  ☆55Updated 3 months ago
• NVIDIA / nsight-training
  Training material for Nsight developer tools
  ☆163Updated last year
• Infrawaves / DeepEP_ibrc_dual-ports_multiQP
  Aims to implement dual-port and multi-qp solutions in deepEP ibrc transport
  ☆61Updated 3 months ago