Alternatives and similar repositories for torchair

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

• CalvinXKY / mfu_calculation
  A simple calculation for LLM MFU.
  ☆50Updated 3 months ago
• Ascend / triton-ascend
  Triton adapter for Ascend. Mirror of https://gitee.com/ascend/triton-ascend
  ☆90Updated this week
• InternLM / turbomind
  ☆97Updated 8 months ago
• flashinfer-ai / cutlass-viz
  ☆65Updated 7 months ago
• AlibabaPAI / FLASHNN
  ☆102Updated last year
• feifeibear / LLMRoofline
  Compare different hardware platforms via the Roofline Model for LLM inference tasks.
  ☆119Updated last year
• DeepLink-org / DLSlime
  DLSlime: Flexible & Efficient Heterogeneous Transfer Toolkit
  ☆82Updated this week
• DeepLink-org / DLBlas
  DLBlas: clean and efficient kernels
  ☆27Updated this week
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆112Updated last year
• flagos-ai / libtriton_jit
  A Triton JIT runtime and ffi provider in C++
  ☆29Updated 2 weeks ago
• LeiWang1999 / Stream-k.tvm
  ☆19Updated last year
• infinigence / Semi-PD
  A prefill & decode disaggregated LLM serving framework with shared GPU memory and fine-grained compute isolation.
  ☆118Updated 6 months ago
• madsys-dev / deepseekv2-profile
  ☆153Updated 9 months ago
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆137Updated 7 months ago
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆90Updated 2 months ago
• KuangjuX / NVSHMEM-Tutorial
  NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer
  ☆146Updated 2 months ago
• gogongxt / nano-sglang
  ☆78Updated 2 weeks ago
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆41Updated 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.
  ☆46Updated 6 months ago
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆73Updated 7 months ago
• vllm-project / flash-attention
  Fast and memory-efficient exact attention
  ☆104Updated this week
• microsoft / AttentionEngine
  ☆114Updated 6 months ago
• inclusionAI / asystem-amem
  A NCCL extension library, designed to efficiently offload GPU memory allocated by the NCCL communication library.
  ☆58Updated 2 weeks 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.
  ☆43Updated 9 months ago
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆104Updated 5 months ago
• infinigence / FlashOverlap
  A lightweight design for computation-communication overlap.
  ☆194Updated 2 months ago
• AlibabaPAI / torchacc
  PyTorch distributed training acceleration framework
  ☆54Updated 4 months ago
• antgroup / DeepXTrace
  DeepXTrace is a lightweight tool for precisely diagnosing slow ranks in DeepEP-based environments.
  ☆73Updated this week
• flashinfer-ai / cubloaty
  a size profiler for cuda binary
  ☆66Updated 2 months ago
• tlc-pack / cutlass_fpA_intB_gemm
  A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer
  ☆96Updated 3 months ago