zhiyuan1i / TorchRWKVLinks
RWKV6 in native pytorch and triton:)
☆11Updated last year
Alternatives and similar repositories for TorchRWKV
Users that are interested in TorchRWKV are comparing it to the libraries listed below
Sorting:
- ☆24Updated last week
- A large-scale RWKV v6, v7(World, PRWKV, Hybrid-RWKV) inference. Capable of inference by combining multiple states(Pseudo MoE). Easy to de…☆40Updated last week
- State tuning tunes the state☆35Updated 5 months ago
- Here we will test various linear attention designs.☆62Updated last year
- Official Code Repository for the paper "Key-value memory in the brain"☆27Updated 5 months ago
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Updated last year
- Implementation of a Light Recurrent Unit in Pytorch☆48Updated 10 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆89Updated last year
- Implementation of 2-simplicial attention proposed by Clift et al. (2019) and the recent attempt to make practical in Fast and Simplex, Ro…☆42Updated 3 weeks ago
- My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated☆33Updated 11 months ago
- Griffin MQA + Hawk Linear RNN Hybrid☆88Updated last year
- ☆23Updated 10 months ago
- JAX Scalify: end-to-end scaled arithmetics☆16Updated 9 months ago
- RADLADS training code☆26Updated 3 months ago
- Explorations into adversarial losses on top of autoregressive loss for language modeling☆37Updated 5 months ago
- GoldFinch and other hybrid transformer components☆46Updated last year
- https://x.com/BlinkDL_AI/status/1884768989743882276☆28Updated 3 months ago
- RWKV-7 mini☆11Updated 4 months ago
- ☆32Updated last year
- A converter and basic tester for rwkv onnx☆42Updated last year
- Efficient implementations of state-of-the-art linear attention models in Pytorch and Triton☆41Updated last week
- RWKV model implementation☆38Updated 2 years ago
- RWKV-7: Surpassing GPT☆94Updated 8 months ago
- Fast modular code to create and train cutting edge LLMs☆67Updated last year
- Triton implement of bi-directional (non-causal) linear attention☆52Updated 6 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…☆46Updated 3 weeks ago
- Some preliminary explorations of Mamba's context scaling.☆14Updated 7 months ago
- ☆32Updated last year
- Implementation of Strassen attention, from Kozachinskiy et al. of National Center of AI in Chile☆41Updated last month
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year