ROCm / TransformerEngine

Related projects ⓘ

Alternatives and complementary repositories for TransformerEngine

• microsoft / DeepSpeed-Kernels
  ☆55Updated 5 months ago
• vllm-project / flash-attention
  Fast and memory-efficient exact attention
  ☆28Updated 2 weeks ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆53Updated last week
• INT-FlashAttention2024 / INT-FlashAttention
  ☆46Updated last month
• triton-lang / kernels
  ☆43Updated last week
• feifeibear / ChituAttention
  Quantized Attention on GPU
  ☆29Updated last week
• HandH1998 / QQQ
  QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.
  ☆76Updated last month
• Azure / msccl-executor-nccl
  ☆33Updated 2 months ago
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆196Updated 2 weeks ago
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆35Updated 6 months ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆87Updated 4 months ago
• AlibabaPAI / FLASHNN
  ☆79Updated 2 months ago
• HanGuo97 / flute
  Fast Matrix Multiplications for Lookup Table-Quantized LLMs
  ☆184Updated last month
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆35Updated 6 months ago
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆156Updated last year
• chengzeyi / ParaAttention
  [WIP] Context parallel attention that works with torch.compile
  ☆20Updated last week
• mit-han-lab / patch_conv
  Patch convolution to avoid large GPU memory usage of Conv2D
  ☆79Updated 5 months ago
• mlc-ai / xgrammar
  ☆22Updated this week
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆16Updated this week
• tlc-pack / cutlass_fpA_intB_gemm
  A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer
  ☆85Updated 8 months ago
• ruikangliu / FlatQuant
  Official PyTorch implementation of FlatQuant: Flatness Matters for LLM Quantization
  ☆59Updated this week
• ColfaxResearch / cfx-article-src
  ☆47Updated 2 weeks ago
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆160Updated last week
• mobiusml / gemlite
  Simple and fast low-bit matmul kernels in CUDA / Triton
  ☆140Updated this week
• IST-DASLab / Sparse-Marlin
  Boosting 4-bit inference kernels with 2:4 Sparsity
  ☆51Updated 2 months ago
• cchan / tccl
  extensible collectives library in triton
  ☆65Updated last month
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆96Updated this week
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆98Updated 2 months ago
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆19Updated 3 weeks ago
• vllm-project / dashboard
  vLLM performance dashboard
  ☆18Updated 6 months ago