zhiyuan1i / TorchRWKVLinks
RWKV6 in native pytorch and triton:)
☆11Updated 10 months ago
Alternatives and similar repositories for TorchRWKV
Users that are interested in TorchRWKV are comparing it to the libraries listed below
Sorting:
- A large-scale RWKV v6, v7(World, PRWKV, Hybrid-RWKV) inference. Capable of inference by combining multiple states(Pseudo MoE). Easy to de…☆35Updated last week
- ☆31Updated last year
- Efficient implementations of state-of-the-art linear attention models in Pytorch and Triton☆29Updated last week
- RWKV-X is a Linear Complexity Hybrid Language Model based on the RWKV architecture, integrating Sparse Attention to improve the model's l…☆37Updated last month
- ☆32Updated last year
- ☆17Updated last month
- Implementation of a Light Recurrent Unit in Pytorch☆47Updated 7 months ago
- Official Code Repository for the paper "Key-value memory in the brain"☆26Updated 3 months ago
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆23Updated 2 weeks ago
- Here we will test various linear attention designs.☆58Updated last year
- ☆23Updated 8 months ago
- RADLADS training code☆22Updated 3 weeks ago
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Updated 11 months ago
- My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated☆32Updated 9 months ago
- State tuning tunes the state☆33Updated 3 months ago
- continous batching and parallel acceleration for RWKV6☆24Updated 11 months ago
- https://x.com/BlinkDL_AI/status/1884768989743882276☆28Updated last month
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆37Updated last year
- A 20M RWKV v6 can do nonogram☆14Updated 7 months ago
- ☆16Updated 2 years ago
- Let us make Psychohistory (as in Asimov) a reality, and accessible to everyone. Useful for LLM grounding and games / fiction / business /…☆39Updated 2 years ago
- [ICML 2024] Official Repository for the paper "Transformers Get Stable: An End-to-End Signal Propagation Theory for Language Models"☆10Updated 10 months ago
- A converter and basic tester for rwkv onnx☆41Updated last year
- ☆49Updated last year
- Triton implement of bi-directional (non-causal) linear attention☆48Updated 4 months ago
- ☆22Updated 5 months ago
- Explorations into adversarial losses on top of autoregressive loss for language modeling☆36Updated 3 months ago
- Inference RWKV with multiple supported backends.☆48Updated this week
- Combining SOAP and MUON☆16Updated 3 months ago
- BigKnow2022: Bringing Language Models Up to Speed☆15Updated 2 years ago