Alternatives and similar repositories for AITemplate:

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

• ROCm / TransformerEngine
  ☆29Updated this week
• ROCm / flash-attention
  Fast and memory-efficient exact attention
  ☆171Updated this week
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆57Updated last week
• ROCm / rocmProfileData
  ☆22Updated 2 months ago
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆118Updated this week
• Repeerc / flash-attention-v2-RDNA3-minimal
  a simple Flash Attention v2 implementation with ROCM (RDNA3 GPU, roc wmma), mainly used for stable diffusion(ComfyUI) in Windows ZLUDA en…
  ☆41Updated 8 months ago
• ROCm / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆75Updated this week
• ROCm / aiter
  AI Tensor Engine for ROCm
  ☆180Updated this week
• ColfaxResearch / cfx-article-src
  ☆102Updated last month
• ROCm / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆22Updated last week
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆179Updated last week
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆106Updated 9 months ago
• ColfaxResearch / cutlass-kernels
  ☆198Updated 9 months ago
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆248Updated 5 months ago
• ROCm / composable_kernel
  Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
  ☆383Updated this week
• ROCm / AMDMIGraphX
  AMD's graph optimization engine.
  ☆215Updated this week
• AlibabaPAI / FLASHNN
  ☆92Updated 7 months ago
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆37Updated 9 months ago
• ROCm / rocWMMA
  rocWMMA
  ☆109Updated this week
• ROCm / rocprofiler
  ROC profiler library. Profiling with perf-counters and derived metrics.
  ☆141Updated last week
• ROCm / pytorch-micro-benchmarking
  ☆20Updated 3 weeks ago
• powderluv / vllm-docs
  Documentation for vLLM Dev Channel releases
  ☆9Updated 4 months ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆96Updated 9 months ago
• ROCm / hipBLASLt
  hipBLASLt is a library that provides general matrix-matrix operations with a flexible API and extends functionalities beyond a traditiona…
  ☆91Updated this week
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆261Updated last month
• RRZE-HPC / gpu-benches
  collection of benchmarks to measure basic GPU capabilities
  ☆359Updated 2 months ago
• ROCm / hipBLAS
  ROCm BLAS marshalling library
  ☆138Updated this week
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆40Updated last month
• intel / torch-xpu-ops
  ☆38Updated this week
• yifuwang / symm-mem-recipes
  ☆66Updated 3 months ago