Alternatives and similar repositories for torchair

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

• flashinfer-ai / cutlass-viz
  ☆64Updated 5 months ago
• InternLM / turbomind
  ☆95Updated 6 months ago
• AlibabaPAI / FLASHNN
  ☆98Updated last year
• DeepLink-org / DLSlime
  DLSlime: Flexible & Efficient Heterogeneous Transfer Toolkit
  ☆67Updated this week
• feifeibear / LLMRoofline
  Compare different hardware platforms via the Roofline Model for LLM inference tasks.
  ☆113Updated last year
• Ascend / triton-ascend
  Triton adapter for Ascend. Mirror of https://gitee.com/ascend/triton-ascend
  ☆74Updated this week
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆116Updated 4 months ago
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆70Updated 4 months ago
• CalvinXKY / mfu_calculation
  A simple calculation for LLM MFU.
  ☆45Updated 2 weeks ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆110Updated last year
• tile-ai / tilescale
  Tile-based language built for AI computation across all scales
  ☆59Updated last week
• infinigence / FlashOverlap
  A lightweight design for computation-communication overlap.
  ☆171Updated last week
• KuangjuX / NVSHMEM-Tutorial
  NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer
  ☆128Updated last week
• thunlp / Seq1F1B
  Sequence-level 1F1B schedule for LLMs.
  ☆32Updated last month
• AlibabaPAI / torchacc
  PyTorch distributed training acceleration framework
  ☆52Updated last month
• KuangjuX / AttnLink
  An experimental communicating attention kernel based on DeepEP.
  ☆34Updated last month
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆40Updated last year
• 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.
  ☆43Updated 3 months ago
• FlagTree / libtriton_jit
  A Triton JIT runtime and ffi provider in C++
  ☆25Updated last week
• madsys-dev / deepseekv2-profile
  ☆147Updated 6 months ago
• infinigence / Semi-PD
  A prefill & decode disaggregated LLM serving framework with shared GPU memory and fine-grained compute isolation.
  ☆111Updated 4 months ago
• alibaba / easydist
  Automated Parallelization System and Infrastructure for Multiple Ecosystems
  ☆79Updated 10 months ago
• tlc-pack / cutlass_fpA_intB_gemm
  A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer
  ☆94Updated 2 weeks ago
• Chtholly-Boss / swizzle
  A practical way of learning Swizzle
  ☆28Updated 7 months ago
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆50Updated last year
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆97Updated 3 months ago
• xlite-dev / ffpa-attn
  🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.
  ☆218Updated last month
• TiledTensor / TiledCUDA
  We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …
  ☆186Updated 8 months ago
• microsoft / AttentionEngine
  ☆97Updated 4 months ago
• 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 7 months ago