Alternatives and similar repositories for torchair

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

• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆40Updated last year
• InternLM / turbomind
  ☆92Updated 5 months ago
• tile-ai / tilescale
  Tile-based language built for AI computation across all scales
  ☆47Updated this week
• flashinfer-ai / cutlass-viz
  ☆63Updated 4 months ago
• AlibabaPAI / FLASHNN
  ☆98Updated 11 months ago
• Ascend / triton-ascend
  Triton adapter for Ascend. Mirror of https://gitee.com/ascend/triton-ascend
  ☆67Updated this week
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆110Updated 11 months ago
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆70Updated 3 months ago
• feifeibear / LLMRoofline
  Compare different hardware platforms via the Roofline Model for LLM inference tasks.
  ☆112Updated last year
• 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
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆107Updated 3 months ago
• Bruce-Lee-LY / decoding_attention
  Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.
  ☆40Updated 2 months ago
• infinigence / FlashOverlap
  A lightweight design for computation-communication overlap.
  ☆160Updated last week
• Chtholly-Boss / swizzle
  A practical way of learning Swizzle
  ☆25Updated 6 months ago
• KuangjuX / AttnLink
  An experimental communicating attention kernel based on DeepEP.
  ☆34Updated last month
• deepspeedai / DeepSpeed-Kernels
  ☆74Updated 5 months ago
• 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
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆81Updated 2 weeks ago
• CalvinXKY / mfu_calculation
  A simple calculation for LLM MFU.
  ☆44Updated 5 months ago
• LeiWang1999 / Stream-k.tvm
  ☆19Updated 11 months ago
• microsoft / AttentionEngine
  ☆92Updated 3 months ago
• FlagOpen / FlagCX
  ☆84Updated this week
• Bruce-Lee-LY / flash_attention_inference
  Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.
  ☆40Updated 6 months ago
• vllm-project / flash-attention
  Fast and memory-efficient exact attention
  ☆91Updated this week
• infinigence / Semi-PD
  A prefill & decode disaggregated LLM serving framework with shared GPU memory and fine-grained compute isolation.
  ☆105Updated 3 months ago
• luliyucoordinate / cute-flash-attention
  Implement Flash Attention using Cute.
  ☆95Updated 8 months ago
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆47Updated last year
• madsys-dev / deepseekv2-profile
  ☆146Updated 5 months ago
• antgroup / DeepXTrace
  DeepXTrace is a lightweight tool for precisely diagnosing slow ranks in DeepEP-based environments.
  ☆46Updated 2 weeks ago
• sgl-project / tensorrt-demo
  TensorRT LLM Benchmark Configuration
  ☆13Updated last year