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
  ☆200Updated last month
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆137Updated last week
• ROCm / aiter
  AI Tensor Engine for ROCm
  ☆301Updated 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…
  ☆48Updated last year
• ROCm / rocmProfileData
  ☆27Updated last month
• ROCm / composable_kernel
  Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
  ☆483Updated this week
• ROCm / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆108Updated this week
• ROCm / TransformerEngine
  ☆51Updated this week
• ROCm / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆23Updated last week
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆219Updated this week
• ROCm / TheRock
  The HIP Environment and ROCm Kit - A lightweight open source build system for HIP and ROCm
  ☆563Updated this week
• ROCm / rocm-libraries
  monorepo for rocm libraries
  ☆177Updated this week
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆83Updated last week
• RRZE-HPC / gpu-benches
  collection of benchmarks to measure basic GPU capabilities
  ☆456Updated 3 weeks ago
• ROCm / bitsandbytes
  8-bit CUDA functions for PyTorch
  ☆68Updated last month
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆272Updated 4 months ago
• mk1-project / quickreduce
  QuickReduce is a performant all-reduce library designed for AMD ROCm that supports inline compression.
  ☆35Updated 2 months ago
• ROCm / rccl
  ROCm Communication Collectives Library (RCCL)
  ☆399Updated this week
• ROCm / iris
  AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming
  ☆111Updated this week
• ROCm / AMDMIGraphX
  AMD's graph optimization engine.
  ☆266Updated this week
• microsoft / BitBLAS
  BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.
  ☆709Updated 3 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…
  ☆63Updated 4 months ago
• ROCm / rocWMMA
  [DEPRECATED] Moved to ROCm/rocm-libraries repo
  ☆137Updated this week
• ROCm / amd_matrix_instruction_calculator
  A tool for generating information about the matrix multiplication instructions in AMD Radeon™ and AMD Instinct™ accelerators
  ☆120Updated this week
• intel / torch-xpu-ops
  ☆61Updated this week
• ColfaxResearch / cutlass-kernels
  ☆243Updated last year
• intel / sycl-tla
  SYCL* Templates for Linear Algebra (SYCL*TLA) - SYCL based CUTLASS implementation for Intel GPUs
  ☆51Updated this week
• ROCm / hipBLASLt
  [DEPRECATED] Moved to ROCm/rocm-libraries repo
  ☆114Updated this week
• ColfaxResearch / cfx-article-src
  ☆154Updated 6 months ago
• ROCm / mori
  Modular RDMA Interface
  ☆59Updated this week