ROCm / aotriton

Related projects ⓘ

Alternatives and complementary repositories for aotriton

• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆90Updated 4 months ago
• triton-lang / kernels
  ☆45Updated 2 weeks ago
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆166Updated 3 weeks ago
• ColfaxResearch / cutlass-kernels
  ☆169Updated 4 months ago
• iree-org / iree-nvgpu
  ☆48Updated 8 months ago
• cchan / tccl
  extensible collectives library in triton
  ☆72Updated last month
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆156Updated last year
• mobiusml / gemlite
  Simple and fast low-bit matmul kernels in CUDA / Triton
  ☆145Updated this week
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆143Updated this week
• stanford-futuredata / stk
  ☆88Updated 2 months ago
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆191Updated this week
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆53Updated 3 weeks ago
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆93Updated this week
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆98Updated 2 months ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆271Updated this week
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆209Updated 3 weeks ago
• ColfaxResearch / cfx-article-src
  ☆48Updated this week
• yifuwang / symm-mem-recipes
  ☆12Updated last month
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆59Updated 6 years ago
• microsoft / DeepSpeed-Kernels
  ☆55Updated 5 months ago
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆35Updated 6 months ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆211Updated 3 months ago
• ankan-ban / llama_cu_awq
  llama INT4 cuda inference with AWQ
  ☆48Updated 4 months ago
• tlc-pack / cutlass_fpA_intB_gemm
  A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer
  ☆85Updated 8 months ago
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆81Updated this week
• AlibabaPAI / FLASHNN
  ☆79Updated 2 months ago
• triton-lang / triton-cpu
  An experimental CPU backend for Triton
  ☆56Updated last week
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆60Updated 8 months ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆98Updated last week
• ROCm / TransformerEngine
  ☆13Updated this week
• INT-FlashAttention2024 / INT-FlashAttention
  ☆47Updated 2 months ago