triton-lang / kernels

Related projects ⓘ

Alternatives and complementary repositories for kernels

• cchan / tccl
  extensible collectives library in triton
  ☆61Updated last month
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆51Updated last week
• mobiusml / gemlite
  Simple and fast low-bit matmul kernels in CUDA / Triton
  ☆133Updated this week
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆87Updated 3 months ago
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆159Updated last week
• ColfaxResearch / cutlass-kernels
  ☆162Updated 3 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆63Updated last week
• triton-lang / triton-cpu
  An experimental CPU backend for Triton
  ☆55Updated last week
• AlibabaPAI / FLASHNN
  ☆79Updated 2 months ago
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆196Updated last week
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆107Updated 5 months ago
• stanford-futuredata / stk
  ☆88Updated 2 months ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆140Updated this week
• microsoft / DeepSpeed-Kernels
  ☆55Updated 5 months ago
• IST-DASLab / Sparse-Marlin
  Boosting 4-bit inference kernels with 2:4 Sparsity
  ☆51Updated 2 months ago
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆132Updated 2 months ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆98Updated last month
• ankan-ban / llama_cu_awq
  llama INT4 cuda inference with AWQ
  ☆47Updated 4 months ago
• TiledTensor / TiledCUDA
  TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.
  ☆148Updated this week
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆96Updated this week
• ColfaxResearch / cfx-article-src
  ☆47Updated 2 weeks ago
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆35Updated 6 months ago
• hao-ai-lab / MuxServe
  ☆41Updated 4 months ago
• HanGuo97 / flute
  Fast Matrix Multiplications for Lookup Table-Quantized LLMs
  ☆183Updated last month
• INT-FlashAttention2024 / INT-FlashAttention
  ☆45Updated last month
• microsoft / SparTA
  ☆130Updated 3 months ago
• AlibabaResearch / flash-llm
  Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity
  ☆177Updated last year
• sgl-project / sgl-learning-materials
  Materials for learning SGLang
  ☆75Updated this week
• mit-han-lab / deepcompressor
  ☆121Updated this week
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆34Updated 2 years ago