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
  ☆194Updated last week
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆136Updated this week
• ROCm / aiter
  AI Tensor Engine for ROCm
  ☆292Updated this week
• ROCm / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆107Updated this week
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆80Updated last week
• ROCm / TransformerEngine
  ☆51Updated this week
• ROCm / bitsandbytes
  8-bit CUDA functions for PyTorch
  ☆66Updated last month
• ROCm / composable_kernel
  Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
  ☆478Updated last 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 4 months ago
• ROCm / rocmProfileData
  ☆27Updated last month
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆266Updated 3 months ago
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆215Updated this week
• ROCm / rccl
  ROCm Communication Collectives Library (RCCL)
  ☆390Updated 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 / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆23Updated 3 weeks ago
• mk1-project / quickreduce
  QuickReduce is a performant all-reduce library designed for AMD ROCm that supports inline compression.
  ☆33Updated 2 months ago
• intel / torch-xpu-ops
  ☆60Updated this week
• microsoft / BitBLAS
  BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.
  ☆703Updated 2 months ago
• ROCm / iris
  AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming
  ☆93Updated this week
• ColfaxResearch / cutlass-kernels
  ☆241Updated last year
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆102Updated 2 months ago
• ROCm / hipBLASLt
  [DEPRECATED] Moved to ROCm/rocm-libraries repo
  ☆114Updated this week
• meta-pytorch / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆301Updated 2 months ago
• ROCm / rccl-tests
  RCCL Performance Benchmark Tests
  ☆76Updated last week
• intel / xetla
  ☆62Updated 10 months ago
• intel / sycl-tla
  SYCL* Templates for Linear Algebra (SYCL*TLA) - SYCL based CUTLASS implementation for Intel GPUs
  ☆44Updated this week
• HabanaAI / vllm-fork
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆83Updated last week
• ROCm / AMDMIGraphX
  AMD's graph optimization engine.
  ☆262Updated this week
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆41Updated last year
• dropbox / gemlite
  Fast low-bit matmul kernels in Triton
  ☆385Updated last week