uniartisan / TorchRWKV
RWKV6 in native pytorch and triton:)
☆11Updated 5 months ago
Alternatives and similar repositories for TorchRWKV:
Users that are interested in TorchRWKV are comparing it to the libraries listed below
- Efficient implementations of state-of-the-art linear attention models in Pytorch and Triton☆15Updated this week
- A large-scale RWKV v6, v7 inference. Capable of inference by combining multiple states(Pseudo MoE). Easy to deploy on docker. Supports tr…☆25Updated last week
- State tuning tunes the state☆29Updated 10 months ago
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Updated 7 months ago
- ☆24Updated 4 months ago
- ☆14Updated last month
- HGRN2: Gated Linear RNNs with State Expansion☆52Updated 5 months ago
- My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated☆31Updated 5 months ago
- Official Code Repository for the paper "Key-value memory in the brain"☆20Updated last week
- Implementation of a Light Recurrent Unit in Pytorch☆48Updated 3 months ago
- ☆31Updated 8 months ago
- Here we will test various linear attention designs.☆58Updated 9 months ago
- BigKnow2022: Bringing Language Models Up to Speed☆14Updated last year
- ☆17Updated last month
- ☆16Updated last year
- Official repository for ICML 2024 paper "MoRe Fine-Tuning with 10x Fewer Parameters"☆17Updated 3 weeks ago
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆23Updated last month
- Official code for the paper "Attention as a Hypernetwork"☆23Updated 7 months ago
- ☆33Updated 6 months ago
- RWKV, in easy to read code☆62Updated 2 months ago
- A library for fast and efficient mLSTM Kernels.☆29Updated last month
- ☆13Updated last month
- ☆32Updated last year
- TriNet: stabilizing self-supervised learning from complete or slow collapse on ASR.☆26Updated last year
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆36Updated last year
- Awesome Triton Resources☆19Updated last month
- Triton implement of bi-directional (non-causal) linear attention☆38Updated 2 weeks ago
- A repository for research on medium sized language models.☆76Updated 8 months ago
- Implementing Mamba SSM into a mini language model and training it on the open domain works of Sherlock Holmes. Also, implementation of pa…☆9Updated 10 months ago