Alternatives and similar repositories for TritonBench

Users that are interested in TritonBench are comparing it to the libraries listed below

• flashinfer-ai / cutlass-viz
  ☆65Updated 6 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆125Updated 5 months ago
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆84Updated last month
• triton-lang / kernels
  ☆93Updated 11 months ago
• tile-ai / AttentionEngine
  ☆50Updated 5 months ago
• ByteDance-Seed / cudaLLM
  ☆120Updated 2 months ago
• luliyucoordinate / cute-flash-attention
  Implement Flash Attention using Cute.
  ☆96Updated 10 months ago
• HanGuo97 / hilt
  ☆35Updated last week
• feifeibear / DPSKV3MFU
  Estimate MFU for DeepSeekV3
  ☆26Updated 9 months ago
• microsoft / AttentionEngine
  ☆102Updated 5 months ago
• andy-yang-1 / DoubleSparse
  16-fold memory access reduction with nearly no loss
  ☆104Updated 7 months ago
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆100Updated 4 months ago
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆72Updated 5 months ago
• DD-DuDa / BitLadder
  A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.
  ☆60Updated last week
• fzyzcjy / torch_utils
  Utility scripts for PyTorch (e.g. Memory profiler that understands more low-level allocations such as NCCL)
  ☆62Updated last month
• microsoft / SparTA
  ☆154Updated last year
• IST-DASLab / qutlass
  QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning
  ☆120Updated last week
• ByteDance-Seed / FlexPrefill
  Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference
  ☆148Updated 2 weeks ago
• INT-FlashAttention2024 / INT-FlashAttention
  ☆82Updated 9 months ago
• zhuzilin / flash-attention-with-sink
  ☆39Updated 2 months ago
• meta-pytorch / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆272Updated this week
• AlibabaResearch / flash-llm
  Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity
  ☆221Updated 2 years ago
• toyaix / triton-runner
  Multi-Level Triton Runner supporting Python, IR, PTX, and cubin.
  ☆75Updated last week
• FasterDecoding / TEAL
  ☆145Updated 8 months ago
• stanford-futuredata / stk
  ☆112Updated last year
• d-matrix-ai / keyformer-llm
  ☆58Updated last year
• opengear-project / GEAR
  GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM
  ☆169Updated last year
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆117Updated last year
• PKU-SEC-Lab / HybriMoE
  [DAC'25] Official implement of "HybriMoE: Hybrid CPU-GPU Scheduling and Cache Management for Efficient MoE Inference"
  ☆81Updated 4 months ago
• tile-ai / tilescale
  Tile-based language built for AI computation across all scales
  ☆74Updated this week