chengzeyi / ParaAttention

Related projects ⓘ

Alternatives and complementary repositories for ParaAttention

• feifeibear / ChituAttention
  Quantized Attention on GPU
  ☆29Updated last week
• chengzeyi / piflux
  (WIP) Parallel inference for black-forest-labs' FLUX model.
  ☆4Updated last week
• mit-han-lab / patch_conv
  Patch convolution to avoid large GPU memory usage of Conv2D
  ☆79Updated 5 months ago
• xdit-project / DistVAE
  A parallelism VAE avoids OOM for high resolution image generation
  ☆40Updated last month
• INT-FlashAttention2024 / INT-FlashAttention
  ☆46Updated last month
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆32Updated 3 months ago
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆55Updated 4 months ago
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆20Updated this week
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆35Updated 6 months ago
• sgl-project / tensorrt-demo
  TensorRT LLM Benchmark Configuration
  ☆11Updated 3 months ago
• NVlabs / COAT
  ☆24Updated last week
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆98Updated 2 months ago
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆53Updated last week
• InternLM / turbomind
  ☆27Updated this week
• megvii-research / IntLLaMA
  IntLLaMA: A fast and light quantization solution for LLaMA
  ☆18Updated last year
• ChenMnZ / PrefixQuant
  An algorithm for static activation quantization of LLMs
  ☆68Updated this week
• thu-nics / DiTFastAttn
  ☆97Updated last month
• AlibabaPAI / FLASHNN
  ☆79Updated 2 months ago
• mit-han-lab / Block-Sparse-Attention
  A sparse attention kernel supporting mix sparse patterns
  ☆53Updated 3 weeks ago
• microsoft / DeepSpeed-Kernels
  ☆55Updated 5 months ago
• ruikangliu / FlatQuant
  Official PyTorch implementation of FlatQuant: Flatness Matters for LLM Quantization
  ☆59Updated this week
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆87Updated 4 months ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last month
• flashinfer-ai / debug-print
  Debug print operator for cudagraph debugging
  ☆10Updated 3 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
• Juanerx / Q-DiT
  PyTorch code for Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers
  ☆33Updated 2 months ago
• mit-han-lab / deepcompressor
  Model Compression Toolbox for Large Language Models and Diffusion Models
  ☆188Updated this week
• thu-ml / Jetfire-INT8Training
  ☆21Updated 3 months ago
• Infini-AI-Lab / MagicDec
  Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding
  ☆70Updated last week