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
  ☆191Updated this week
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆133Updated this week
• ROCm / aiter
  AI Tensor Engine for ROCm
  ☆284Updated this week
• ROCm / TransformerEngine
  ☆47Updated this week
• ROCm / composable_kernel
  Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
  ☆470Updated this week
• ROCm / rocmProfileData
  ☆27Updated last week
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆210Updated this week
• ROCm / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆103Updated this week
• 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 3 months ago
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆77Updated this week
• mk1-project / quickreduce
  QuickReduce is a performant all-reduce library designed for AMD ROCm that supports inline compression.
  ☆33Updated last month
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆265Updated 2 months ago
• intel / sycl-tla
  SYCL* Templates for Linear Algebra (SYCL*TLA) - SYCL based CUTLASS implementation for Intel GPUs
  ☆40Updated this week
• ROCm / rocm-libraries
  monorepo for rocm libraries
  ☆132Updated this week
• ROCm / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆23Updated this week
• ROCm / AMDMIGraphX
  AMD's graph optimization engine.
  ☆253Updated 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 / bitsandbytes
  8-bit CUDA functions for PyTorch
  ☆63Updated 2 weeks ago
• ColfaxResearch / cutlass-kernels
  ☆238Updated last year
• ColfaxResearch / cfx-article-src
  ☆148Updated 5 months ago
• intel / torch-xpu-ops
  ☆57Updated this week
• ROCm / rocWMMA
  rocWMMA
  ☆133Updated this week
• ROCm / hipBLASLt
  [DEPRECATED] Moved to ROCm/rocm-libraries repo
  ☆111Updated this week
• ROCm / rccl
  ROCm Communication Collectives Library (RCCL)
  ☆386Updated this week
• microsoft / BitBLAS
  BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.
  ☆685Updated 2 months ago
• intel / xetla
  ☆63Updated 9 months ago
• RRZE-HPC / gpu-benches
  collection of benchmarks to measure basic GPU capabilities
  ☆427Updated 7 months ago
• NVIDIA / cudnn-frontend
  cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it
  ☆623Updated 3 weeks ago
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆102Updated 2 months ago
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆112Updated last week