BlinkDL / RWKV-CUDA
The CUDA version of the RWKV language model ( https://github.com/BlinkDL/RWKV-LM )
☆220Updated 3 months ago
Alternatives and similar repositories for RWKV-CUDA:
Users that are interested in RWKV-CUDA are comparing it to the libraries listed below
- GPTQ inference Triton kernel☆298Updated last year
- ☆157Updated last year
- The official implementation of the EMNLP 2023 paper LLM-FP4☆191Updated last year
- 📚FFPA(Split-D): Yet another Faster Flash Prefill Attention with O(1) GPU SRAM complexity for headdim > 256, ~2x↑🎉vs SDPA EA.☆147Updated last week
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆298Updated 8 months ago
- Reorder-based post-training quantization for large language model☆185Updated last year
- This repository contains integer operators on GPUs for PyTorch.☆196Updated last year
- A collection of memory efficient attention operators implemented in the Triton language.☆253Updated 9 months ago
- ☆145Updated last year
- Code repo for the paper "LLM-QAT Data-Free Quantization Aware Training for Large Language Models"☆278Updated 2 weeks ago
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆272Updated last month
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆240Updated 4 months ago
- Microsoft Automatic Mixed Precision Library☆581Updated 5 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆106Updated 6 months ago
- flash attention tutorial written in python, triton, cuda, cutlass☆302Updated 2 months ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆101Updated 8 months ago
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆471Updated last year
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆35Updated 3 weeks ago
- PyTorch bindings for CUTLASS grouped GEMM.☆74Updated 4 months ago
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆557Updated last month
- Applied AI experiments and examples for PyTorch☆249Updated this week
- [NeurIPS 2024] KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantization☆336Updated 7 months ago
- ☆74Updated 3 months ago
- ☆125Updated 2 weeks ago
- This repository contains the experimental PyTorch native float8 training UX☆222Updated 7 months ago
- Zero Bubble Pipeline Parallelism☆373Updated 2 weeks ago
- PyTorch bindings for CUTLASS grouped GEMM.☆108Updated 2 months ago
- [ICML 2024] SqueezeLLM: Dense-and-Sparse Quantization☆680Updated 7 months ago
- ☆139Updated 11 months ago
- A model compression and acceleration toolbox based on pytorch.☆329Updated last year