BlinkDL / RWKV-CUDALinks
The CUDA version of the RWKV language model ( https://github.com/BlinkDL/RWKV-LM )
☆222Updated 6 months ago
Alternatives and similar repositories for RWKV-CUDA
Users that are interested in RWKV-CUDA are comparing it to the libraries listed below
Sorting:
- Reorder-based post-training quantization for large language model☆191Updated 2 years ago
- GPTQ inference Triton kernel☆302Updated 2 years ago
- The official implementation of the EMNLP 2023 paper LLM-FP4☆206Updated last year
- ☆157Updated last year
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆252Updated 7 months ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆109Updated 11 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆272Updated last year
- Standalone Flash Attention v2 kernel without libtorch dependency☆110Updated 9 months ago
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆310Updated 11 months ago
- This repository contains integer operators on GPUs for PyTorch.☆205Updated last year
- ⚡️FFPA: Extend FlashAttention-2 with Split-D, achieve ~O(1) SRAM complexity for large headdim, 1.8x~3x↑ vs SDPA.☆186Updated last month
- ☆124Updated last year
- ☆139Updated last year
- ☆86Updated 2 months ago
- ☆127Updated 5 months ago
- [NeurIPS 2024] KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantization☆358Updated 10 months ago
- Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernels☆106Updated last year
- [ICML 2024] SqueezeLLM: Dense-and-Sparse Quantization☆692Updated 10 months ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 10 months ago
- Code repo for the paper "LLM-QAT Data-Free Quantization Aware Training for Large Language Models"☆295Updated 3 months ago
- Zero Bubble Pipeline Parallelism☆398Updated last month
- Code for the paper "QMoE: Practical Sub-1-Bit Compression of Trillion-Parameter Models".☆277Updated last year
- A general 2-8 bits quantization toolbox with GPTQ/AWQ/HQQ/VPTQ, and export to onnx/onnx-runtime easily.☆172Updated 2 months ago
- Low-bit optimizers for PyTorch☆128Updated last year
- Microsoft Automatic Mixed Precision Library☆609Updated 8 months ago
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆313Updated 4 months ago
- ☆194Updated last month
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆629Updated last month
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆126Updated 2 months ago
- Applied AI experiments and examples for PyTorch☆277Updated 3 weeks ago