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:
- Implementation of a Light Recurrent Unit in Pytorch☆49Updated 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…☆47Updated 3 months ago
- Some preliminary explorations of Mamba's context scaling.☆13Updated 11 months ago
- ☆27Updated 4 months ago
- https://x.com/BlinkDL_AI/status/1884768989743882276☆28Updated 7 months ago
- Explorations into adversarial losses on top of autoregressive loss for language modeling☆38Updated 9 months ago
- Here we will test various linear attention designs.☆62Updated last year
- State tuning tunes the state☆35Updated 9 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 last month
- 32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.☆49Updated 2 years ago
- ☆32Updated last year
- Griffin MQA + Hawk Linear RNN Hybrid☆89Updated last year
- Fast modular code to create and train cutting edge LLMs☆68Updated last year
- RWKV-7 mini☆11Updated 8 months ago
- RWKV-7: Surpassing GPT☆101Updated last year
- Let us make Psychohistory (as in Asimov) a reality, and accessible to everyone. Useful for LLM grounding and games / fiction / business /…☆40Updated 2 years ago
- My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated☆33Updated last year
- ☆39Updated 7 months ago
- JAX Scalify: end-to-end scaled arithmetics☆17Updated last year
- Course Project for COMP4471 on RWKV☆17Updated last year
- GoldFinch and other hybrid transformer components☆45Updated last year
- MiSS is a novel PEFT method that features a low-rank structure but introduces a new update mechanism distinct from LoRA, achieving an exc…☆25Updated last month
- Implementation of GateLoop Transformer in Pytorch and Jax☆91Updated last year
- Official Code Repository for the paper "Key-value memory in the brain"☆29Updated 9 months ago
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆120Updated last year
- RWKV, in easy to read code☆72Updated 8 months ago
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Updated last year
- Implementation of Strassen attention, from Kozachinskiy et al. of National Center of AI in Chile☆41Updated 4 months ago
- HGRN2: Gated Linear RNNs with State Expansion☆55Updated last year
- Triton implement of bi-directional (non-causal) linear attention☆56Updated 10 months ago