Alternatives and similar repositories for AITemplate:

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

• ROCm / flash-attention
  Fast and memory-efficient exact attention
  ☆159Updated this week
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆165Updated this week
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆54Updated this week
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆108Updated this week
• ROCm / rocmProfileData
  ☆19Updated last week
• ROCm / composable_kernel
  Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
  ☆355Updated this week
• ROCm / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆67Updated this week
• ROCm / AMDMIGraphX
  AMD's graph optimization engine.
  ☆210Updated 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…
  ☆37Updated 6 months ago
• intel / intel-extension-for-deepspeed
  Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…
  ☆61Updated 2 months ago
• ROCm / TransformerEngine
  ☆20Updated this week
• ROCm / pytorch-micro-benchmarking
  ☆19Updated 3 months ago
• intel / torch-xpu-ops
  ☆34Updated this week
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆89Updated 2 months ago
• ROCm / rccl-tests
  RCCL Performance Benchmark Tests
  ☆59Updated last month
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆237Updated 4 months ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆100Updated 7 months ago
• ColfaxResearch / cutlass-kernels
  ☆185Updated 7 months ago
• ColfaxResearch / cfx-article-src
  ☆75Updated last week
• ROCm / rccl
  ROCm Communication Collectives Library (RCCL)
  ☆303Updated this week
• libxsmm / tpp-pytorch-extension
  Intel® Tensor Processing Primitives extension for Pytorch*
  ☆11Updated this week
• ROCm / rocm-examples
  A collection of examples for the ROCm software stack
  ☆186Updated this week
• RRZE-HPC / gpu-benches
  collection of benchmarks to measure basic GPU capabilities
  ☆303Updated 2 weeks ago
• triton-lang / kernels
  ☆71Updated 3 months ago
• intel / xetla
  ☆60Updated 2 months ago
• yifuwang / symm-mem-recipes
  ☆48Updated 2 months ago
• ROCm / pyrsmi
  python package of rocm-smi-lib
  ☆20Updated 5 months ago
• NVIDIA / cudnn-frontend
  cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it
  ☆508Updated last month
• amd / ZenDNN
  ☆105Updated 3 months ago