Alternatives and similar repositories for FlashMLA

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

• infinigence / Semi-PD
  A prefill & decode disaggregated LLM serving framework with shared GPU memory and fine-grained compute isolation.
  ☆87Updated last month
• OpenPPL / CuAssembler
  An unofficial cuda assembler, for all generations of SASS, hopefully ：）
  ☆83Updated 2 years ago
• 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.
  ☆35Updated 5 months ago
• FlagOpen / FlagCX
  ☆66Updated last week
• GVProf / GVProf
  GVProf: A Value Profiler for GPU-based Clusters
  ☆50Updated last year
• pigirons / conv3x3_m1
  This is a demo how to write a high performance convolution run on apple silicon
  ☆54Updated 3 years ago
• tlc-pack / cutlass_fpA_intB_gemm
  A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer
  ☆92Updated 3 weeks ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆99Updated 11 months ago
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆79Updated last month
• yester31 / Cutlass_EX
  study of cutlass
  ☆21Updated 7 months ago
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆90Updated 2 weeks ago
• UDC-GAC / venom
  A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores
  ☆51Updated last year
• BBuf / tensorrt-llm-moe
  ☆29Updated 4 months ago
• Mengjintao / FastCNN
  ☆20Updated 2 years ago
• FlagTree / flagtree
  FlagTree is a unified compiler for multiple AI chips, which is forked from triton-lang/triton.
  ☆43Updated this week
• LeiWang1999 / tvm_gpu_gemm
  play gemm with tvm
  ☆91Updated last year
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆110Updated 9 months ago
• thu-pacman / PET
  PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections
  ☆121Updated 2 years ago
• sjfeng1999 / gpu-arch-microbenchmark
  Dissecting NVIDIA GPU Architecture
  ☆96Updated 2 years ago
• heheda12345 / MagPy
  ☆39Updated last year
• billmuch / matmul_perf_test
  ☆14Updated 3 years ago
• bytedance / ByteMLPerf
  AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…
  ☆246Updated last week
• QianyanTech / NBAssembler
  Assembler and Decompiler for NVIDIA (Maxwell Pascal Volta Turing Ampere) GPUs.
  ☆80Updated 2 years ago
• matrix97317 / OneNeuralNetwork
  This is a cross-chip platform collection of operators and a unified neural network library.
  ☆17Updated last year
• Ascend / torchair
  ☆14Updated this week
• ROCm / roctracer
  ROCm Tracer Callback/Activity Library for Performance tracing AMD GPUs
  ☆83Updated this week
• ROCm / MISA
  Machine Intelligence Shader Autogen. AMDGPU ML shader code generator. (previously iGEMMgen)
  ☆34Updated last week
• Ascend / triton-ascend
  Triton adapter for Ascend. Mirror of https://gitee.com/ascend/triton-ascend
  ☆54Updated this week
• UofT-EcoSystem / DietCode
  DietCode Code Release
  ☆64Updated 2 years ago
• sunlex0717 / DissectingTensorCores
  ☆98Updated last year