Alternatives and similar repositories for learn_torch.compile

Users that are interested in learn_torch.compile are comparing it to the libraries listed below

• feifeibear / ChituAttention
  Quantized Attention on GPU
  ☆44Updated last year
• microsoft / AttentionEngine
  ☆118Updated 8 months ago
• tile-ai / AttentionEngine
  ☆52Updated 8 months ago
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆79Updated last year
• IBM / triton-dejavu
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆96Updated 4 months ago
• ByteDance-Seed / cudaLLM
  ☆130Updated 5 months ago
• TiledTensor / TiledBench
  Benchmark tests supporting the TiledCUDA library.
  ☆18Updated last year
• Bruce-Lee-LY / decoding_attention
  Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.
  ☆46Updated 8 months ago
• megvii-research / IntLLaMA
  IntLLaMA: A fast and light quantization solution for LLaMA
  ☆18Updated 2 years ago
• ademeure / DeeperGEMM
  DeeperGEMM: crazy optimized version
  ☆73Updated 9 months ago
• Dao-AILab / gemm-cublas
  ☆22Updated 9 months ago
• li-plus / flash-preference
  Accelerate LLM preference tuning via prefix sharing with a single line of code
  ☆51Updated 7 months ago
• sgl-project / DeepGEMM
  DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling
  ☆21Updated last month
• KuangjuX / AttnLink
  An experimental communicating attention kernel based on DeepEP.
  ☆35Updated 6 months ago
• xdit-project / DiTCacheAnalysis
  An auxiliary project analysis of the characteristics of KV in DiT Attention.
  ☆32Updated last year
• flashinfer-ai / debug-print
  Debug print operator for cudagraph debugging
  ☆14Updated last year
• mit-han-lab / tinychat-tutorial
  ☆77Updated last year
• flashinfer-ai / cutlass-viz
  ☆65Updated 9 months ago
• thib-s / flash-newton-schulz
  My attempt to improve the speed of the newton schulz algorithm, starting from the dion implementation.
  ☆29Updated 2 months ago
• mit-han-lab / patch_conv
  Patch convolution to avoid large GPU memory usage of Conv2D
  ☆95Updated last year
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆56Updated last year
• tridao / flash-attention-wheels
  ☆61Updated 2 years ago
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆30Updated 2 weeks ago
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆40Updated last year
• cherichy / tilecute
  ☆32Updated 7 months ago
• flagos-ai / libtriton_jit
  A Triton JIT runtime and ffi provider in C++
  ☆31Updated 2 weeks ago
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆106Updated 7 months ago
• flashinfer-ai / cubloaty
  a size profiler for cuda binary
  ☆72Updated 3 weeks ago
• Infini-AI-Lab / vortex_torch
  Vortex: A Flexible and Efficient Sparse Attention Framework
  ☆45Updated 3 weeks ago
• tile-ai / tvm
  Open deep learning compiler stack for cpu, gpu and specialized accelerators
  ☆19Updated this week