BlinkDL / RWKV-CUDA

Related projects ⓘ

Alternatives and complementary repositories for RWKV-CUDA

• FlagOpen / FlagAttention
  A collection of memory efficient attention operators implemented in the Triton language.
  ☆219Updated 5 months ago
• jundaf2 / INT8-Flash-Attention-FMHA-Quantization
  ☆156Updated last year
• hahnyuan / RPTQ4LLM
  Reorder-based post-training quantization for large language model
  ☆181Updated last year
• Guangxuan-Xiao / torch-int
  This repository contains integer operators on GPUs for PyTorch.
  ☆183Updated last year
• xijiu9 / Train_Transformers_with_INT4
  ☆134Updated last year
• 66RING / tiny-flash-attention
  flash attention tutorial written in python, triton, cuda, cutlass
  ☆209Updated 5 months ago
• fpgaminer / GPTQ-triton
  GPTQ inference Triton kernel
  ☆284Updated last year
• usyd-fsalab / fp6_llm
  An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).
  ☆209Updated 3 weeks ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆90Updated 4 months ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆98Updated 2 months ago
• OpenPPL / ppl.nn.llm
  ☆140Updated 6 months ago
• microsoft / BitBLAS
  BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.
  ☆420Updated this week
• KnowingNothing / MatmulTutorial
  A Easy-to-understand TensorOp Matmul Tutorial
  ☆292Updated 2 months ago
• megvii-research / Sparsebit
  A model compression and acceleration toolbox based on pytorch.
  ☆327Updated 10 months ago
• thu-ml / low-bit-optimizers
  Low-bit optimizers for PyTorch
  ☆119Updated last year
• FlagOpen / FlagGems
  FlagGems is an operator library for large language models implemented in Triton Language.
  ☆343Updated this week
• efeslab / Atom
  [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving
  ☆278Updated 4 months ago
• ColfaxResearch / cutlass-kernels
  ☆169Updated 4 months ago
• Bruce-Lee-LY / flash_attention_inference
  Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.
  ☆29Updated 2 months ago
• openmlsys / openmlsys-cuda
  Tutorials for writing high-performance GPU operators in AI frameworks.
  ☆123Updated last year
• DefTruth / Awesome-Diffusion-Inference
  📒A small curated list of Awesome Diffusion Inference Papers with codes.
  ☆96Updated this week
• nbasyl / LLM-FP4
  The official implementation of the EMNLP 2023 paper LLM-FP4
  ☆167Updated 11 months ago
• Azure / MS-AMP
  Microsoft Automatic Mixed Precision Library
  ☆525Updated last month
• kyegomez / FlashAttention20
  Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernels
  ☆98Updated last year
• facebookresearch / LLM-QAT
  Code repo for the paper "LLM-QAT Data-Free Quantization Aware Training for Large Language Models"
  ☆254Updated 2 months ago
• daquexian / faster-rwkv
  ☆123Updated 11 months ago
• OpenPPL / ppl.llm.serving
  ☆123Updated 2 weeks ago
• FMInference / DejaVu
  ☆289Updated 7 months ago
• mit-han-lab / patch_conv
  Patch convolution to avoid large GPU memory usage of Conv2D
  ☆79Updated 5 months ago
• tpoisonooo / llama.onnx
  LLaMa/RWKV onnx models, quantization and testcase
  ☆353Updated last year