RWKV-Vibe / rwkv-kitLinks
☆22Updated 10 months ago
Alternatives and similar repositories for rwkv-kit
Users that are interested in rwkv-kit are comparing it to the libraries listed below
Sorting:
- Efficient implementations of state-of-the-art linear attention models in Pytorch and Triton☆45Updated 2 months ago
- The WorldRWKV project aims to implement training and inference across various modalities using the RWKV7 architecture. By leveraging diff…☆60Updated 3 weeks ago
- VisualRWKV is the visual-enhanced version of the RWKV language model, enabling RWKV to handle various visual tasks.☆237Updated 5 months ago
- A large-scale RWKV v7(World, PRWKV, Hybrid-RWKV) inference. Capable of inference by combining multiple states(Pseudo MoE). Easy to deploy…☆45Updated 3 weeks ago
- A repository for DenseSSMs☆89Updated 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…☆51Updated 4 months ago
- RWKV, in easy to read code☆72Updated 7 months ago
- This is an inference framework for the RWKV large language model implemented purely in native PyTorch. The official native implementation…☆130Updated last year
- [EMNLP 2024] RWKV-CLIP: A Robust Vision-Language Representation Learner☆143Updated 5 months ago
- Implementation of Griffin from the paper: "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆56Updated 3 weeks ago
- ☆153Updated 2 weeks ago
- HGRN2: Gated Linear RNNs with State Expansion☆55Updated last year
- A simple but robust PyTorch implementation of RetNet from "Retentive Network: A Successor to Transformer for Large Language Models" (http…☆106Updated last year
- Implementation of MoE Mamba from the paper: "MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts" in Pytorch and Ze…☆114Updated 3 weeks ago
- Implementation of the paper: "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆109Updated last week
- Implementation of the proposed DeepCrossAttention by Heddes et al at Google research, in Pytorch☆94Updated 8 months ago
- RWKV-LM-V7(https://github.com/BlinkDL/RWKV-LM) Under Lightning Framework☆46Updated 3 weeks ago
- Griffin MQA + Hawk Linear RNN Hybrid☆89Updated last year
- Triton implement of bi-directional (non-causal) linear attention☆56Updated 9 months ago
- Implementation of Agent Attention in Pytorch☆91Updated last year
- Reinforcement Learning Toolkit for RWKV.(v6,v7,ARWKV) Distillation,SFT,RLHF(DPO,ORPO), infinite context training, Aligning. Exploring the…☆55Updated last month
- ☆49Updated 5 months ago
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear at…☆104Updated last year
- https://x.com/BlinkDL_AI/status/1884768989743882276☆28Updated 6 months ago
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆132Updated 2 weeks ago
- ☆65Updated 7 months ago
- ☆34Updated last year
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆333Updated 8 months ago
- ☆48Updated last year
- A Triton Kernel for incorporating Bi-Directionality in Mamba2☆75Updated 10 months ago