BlinkDL / RWKV-CUDA
The CUDA version of the RWKV language model ( https://github.com/BlinkDL/RWKV-LM )
☆217Updated last month
Alternatives and similar repositories for RWKV-CUDA:
Users that are interested in RWKV-CUDA are comparing it to the libraries listed below
- Reorder-based post-training quantization for large language model☆184Updated last year
- GPTQ inference Triton kernel☆292Updated last year
- The official implementation of the EMNLP 2023 paper LLM-FP4☆176Updated last year
- ☆157Updated last year
- A collection of memory efficient attention operators implemented in the Triton language.☆233Updated 7 months ago
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆249Updated 9 months ago
- ☆140Updated last year
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆232Updated 3 months ago
- This repository contains integer operators on GPUs for PyTorch.☆190Updated last year
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆291Updated 6 months ago
- ☆140Updated 9 months ago
- ☆124Updated last year
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆503Updated this week
- [NeurIPS 2024] KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantization☆327Updated 5 months ago
- ☆127Updated last month
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆94Updated 6 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆99Updated 4 months ago
- A model compression and acceleration toolbox based on pytorch.☆329Updated last year
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆469Updated 10 months ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆34Updated 4 months ago
- FP16xINT4 LLM inference kernel that can achieve near-ideal ~4x speedups up to medium batchsizes of 16-32 tokens.☆690Updated 4 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆89Updated 8 months ago
- ☆59Updated last month
- Zero Bubble Pipeline Parallelism☆317Updated 2 months ago
- Code for paper: "QuIP: 2-Bit Quantization of Large Language Models With Guarantees"☆358Updated 11 months ago
- Implementation of FlashAttention in PyTorch☆129Updated 2 weeks ago
- Code repo for the paper "LLM-QAT Data-Free Quantization Aware Training for Large Language Models"☆266Updated 4 months ago
- Fast Matrix Multiplications for Lookup Table-Quantized LLMs☆221Updated last week
- Microsoft Automatic Mixed Precision Library☆554Updated 4 months ago
- Transformer related optimization, including BERT, GPT☆59Updated last year