Alternatives and similar repositories for kleidiai

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

• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆108Updated 8 months ago
• mit-han-lab / tinychat-tutorial
  ☆65Updated 6 months ago
• BBuf / tensorrt-llm-moe
  ☆28Updated 3 months ago
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆84Updated this week
• Bruce-Lee-LY / cuda_hgemv
  Several optimization methods of half-precision general matrix vector multiplication (HGEMV) using CUDA core.
  ☆61Updated 8 months ago
• ankan-ban / llama_cu_awq
  llama INT4 cuda inference with AWQ
  ☆54Updated 3 months ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆106Updated 10 months ago
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆69Updated last week
• 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.
  ☆36Updated last month
• yuninxia / awesome-gemm
  📚 A curated list of awesome matrix-matrix multiplication (A * B = C) frameworks, libraries and software
  ☆33Updated 2 months ago
• flashinfer-ai / cutlass-viz
  ☆58Updated 2 weeks ago
• Qualcomm-AI-research / FP8-quantization
  ☆146Updated 2 years ago
• Bruce-Lee-LY / cutlass_gemm
  Multiple GEMM operators are constructed with cutlass to support LLM inference.
  ☆18Updated 7 months ago
• INT-FlashAttention2024 / INT-FlashAttention
  ☆68Updated 3 months ago
• ROCm / TransformerEngine
  ☆32Updated this week
• mlc-ai / relax
  ☆156Updated last month
• caijixueIT / CUDA_Learning_for_Freshman
  ☆11Updated 2 months ago
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆91Updated 6 years ago
• InternLM / turbomind
  ☆84Updated last month
• tlc-pack / cutlass_fpA_intB_gemm
  A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer
  ☆92Updated last week
• pigirons / conv3x3_m1
  This is a demo how to write a high performance convolution run on apple silicon
  ☆54Updated 3 years ago
• Chtholly-Boss / swizzle
  A practical way of learning Swizzle
  ☆19Updated 3 months ago
• mlc-ai / mlc-python
  ☆32Updated this week
• microsoft / FractalTensor
  FractalTensor is a programming framework that introduces a novel approach to organizing data in deep neural networks (DNNs) as a list of …
  ☆26Updated 4 months ago
• zeroine / cutlass-cute-sample
  ☆31Updated last year
• xlite-dev / ffpa-attn-mma
  📚FFPA(Split-D): Extend FlashAttention with Split-D for large headdim, O(1) GPU SRAM complexity, 1.8x~3x↑🎉 faster than SDPA EA.
  ☆171Updated last month
• Terapines / AI-Benchmark
  RISCV C and Triton AI-Benchmark
  ☆16Updated 6 months ago
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆75Updated last month
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆185Updated 11 months ago
• li199603 / sgemm_with_cuda
  SGEMM optimization with cuda step by step
  ☆18Updated last year