ROCm / TransformerEngine

Related projects ⓘ

Alternatives and complementary repositories for TransformerEngine

• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆90Updated 4 months ago
• triton-lang / kernels
  ☆49Updated 2 weeks ago
• microsoft / DeepSpeed-Kernels
  ☆55Updated 6 months ago
• cchan / tccl
  extensible collectives library in triton
  ☆72Updated 2 months ago
• ColfaxResearch / cfx-article-src
  ☆50Updated this week
• mobiusml / gemlite
  Simple and fast low-bit matmul kernels in CUDA / Triton
  ☆147Updated this week
• INT-FlashAttention2024 / INT-FlashAttention
  ☆47Updated 2 months ago
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆96Updated this week
• yifuwang / symm-mem-recipes
  ☆12Updated 2 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆53Updated 3 weeks ago
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆41Updated this week
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆211Updated 3 weeks ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆98Updated 2 months ago
• stanford-futuredata / stk
  ☆90Updated 2 months ago
• ROCm / rccl-tests
  RCCL Performance Benchmark Tests
  ☆51Updated last month
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆84Updated 4 months ago
• vllm-project / flash-attention
  Fast and memory-efficient exact attention
  ☆30Updated last month
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆168Updated 3 weeks ago
• AlibabaResearch / flash-llm
  Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity
  ☆181Updated last year
• 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
• ColfaxResearch / cutlass-kernels
  ☆169Updated 4 months ago
• Azure / msccl-executor-nccl
  ☆33Updated 2 months ago
• IntelLabs / FP8-Emulation-Toolkit
  PyTorch extension for emulating FP8 data formats on standard FP32 Xeon/GPU hardware.
  ☆100Updated 11 months ago
• mlcommons / logging
  MLPerf™ logging library
  ☆30Updated this week
• ROCm / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆45Updated this week
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆30Updated 4 months ago
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆35Updated 6 months ago
• thu-ml / Jetfire-INT8Training
  ☆23Updated 4 months ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆98Updated this week