00ffcc / chunkRWKV6Links
continous batching and parallel acceleration for RWKV6
☆22Updated last year
Alternatives and similar repositories for chunkRWKV6
Users that are interested in chunkRWKV6 are comparing it to the libraries listed below
Sorting:
- ☆32Updated last year
- Flash-Linear-Attention models beyond language☆21Updated 5 months ago
- ☆22Updated 2 years ago
- Xmixers: A collection of SOTA efficient token/channel mixers☆28Updated 5 months ago
- ☆27Updated 6 months ago
- Flash Attention in 300-500 lines of CUDA/C++☆36Updated 5 months ago
- ☆58Updated last year
- Here we will test various linear attention designs.☆62Updated last year
- Awesome Triton Resources☆39Updated 9 months ago
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆66Updated last year
- Transformers components but in Triton☆34Updated 9 months ago
- ☆106Updated last year
- ☆51Updated 2 years ago
- RADLADS training code☆36Updated 9 months ago
- Triton implement of bi-directional (non-causal) linear attention☆65Updated last week
- Fast and memory-efficient exact attention☆75Updated 11 months ago
- Stick-breaking attention☆62Updated 7 months ago
- RWKV-X is a Linear Complexity Hybrid Language Model based on the RWKV architecture, integrating Sparse Attention to improve the model's l…☆54Updated 3 weeks ago
- ☆129Updated 8 months ago
- Low-bit optimizers for PyTorch☆138Updated 2 years ago
- Contextual Position Encoding but with some custom CUDA Kernels https://arxiv.org/abs/2405.18719☆22Updated last year
- Open-sourcing code associated with the AAAI-25 paper "On the Expressiveness and Length Generalization of Selective State-Space Models on …☆14Updated 4 months ago
- ☆120Updated last week
- ☆24Updated last month
- Triton implementation of FlashAttention2 that adds Custom Masks.☆167Updated last year
- ☆49Updated 7 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆79Updated last year
- Linear Attention Sequence Parallelism (LASP)☆88Updated last year
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear at…☆104Updated last year
- Efficient PScan implementation in PyTorch☆17Updated 2 years ago