00ffcc / chunkRWKV6Links
continous batching and parallel acceleration for RWKV6
☆23Updated last year
Alternatives and similar repositories for chunkRWKV6
Users that are interested in chunkRWKV6 are comparing it to the libraries listed below
Sorting:
- Flash-Linear-Attention models beyond language☆17Updated last month
- ☆32Updated last year
- ☆22Updated last year
- RADLADS training code☆28Updated 4 months ago
- ☆105Updated last year
- ☆26Updated 2 months ago
- Here we will test various linear attention designs.☆62Updated last year
- ☆56Updated last year
- ☆129Updated 4 months ago
- ☆49Updated last year
- RWKV-X is a Linear Complexity Hybrid Language Model based on the RWKV architecture, integrating Sparse Attention to improve the model's l…☆47Updated 2 months ago
- Linear Attention Sequence Parallelism (LASP)☆87Updated last year
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆66Updated last year
- Odysseus: Playground of LLM Sequence Parallelism☆77Updated last year
- Stick-breaking attention☆60Updated 3 months ago
- Transformers components but in Triton☆34Updated 4 months ago
- Fast and memory-efficient exact attention☆70Updated 7 months ago
- Contextual Position Encoding but with some custom CUDA Kernels https://arxiv.org/abs/2405.18719☆22Updated last year
- ☆119Updated 4 months ago
- ☆25Updated 2 months ago
- Triton implementation of FlashAttention2 that adds Custom Masks.☆135Updated last year
- Awesome Triton Resources☆36Updated 5 months ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆189Updated 3 months ago
- SQUEEZED ATTENTION: Accelerating Long Prompt LLM Inference☆54Updated 10 months ago
- ☆21Updated 6 months ago
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆24Updated 3 months ago
- ☆101Updated 2 weeks ago
- ☆142Updated 7 months ago
- HALO: Hadamard-Assisted Low-Precision Optimization and Training method for finetuning LLMs. 🚀 The official implementation of https://arx…☆23Updated 7 months ago
- Open-sourcing code associated with the AAAI-25 paper "On the Expressiveness and Length Generalization of Selective State-Space Models on …☆15Updated 2 weeks ago