uniartisan / TorchRWKV
RWKV6 in native pytorch and triton:)
☆10Updated 3 months ago
Related projects ⓘ
Alternatives and complementary repositories for TorchRWKV
- State tuning tunes the state☆26Updated 7 months ago
- Implementation of the LDP module block in PyTorch and Zeta from the paper: "MobileVLM: A Fast, Strong and Open Vision Language Assistant …☆14Updated 8 months ago
- Efficient implementations of state-of-the-art linear attention models in Pytorch and Triton☆13Updated this week
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Updated 5 months ago
- Here we will test various linear attention designs.☆56Updated 6 months ago
- ☆29Updated 5 months ago
- Implementation of a Light Recurrent Unit in Pytorch☆46Updated last month
- ☆21Updated last month
- Zeta implementation of a reusable and plug in and play feedforward from the paper "Exponentially Faster Language Modeling"☆15Updated this week
- Implementation of Spectral State Space Models☆17Updated 8 months ago
- continous batching and parallel acceleration for RWKV6☆23Updated 4 months ago
- My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated☆30Updated 2 months ago
- A large-scale RWKV v6 inference with FLA . Capable of inference by combining multiple states(Pseudo MoE). Easy to deploy on docker. Suppo…☆16Updated this week
- ☆15Updated last year
- A converter and basic tester for rwkv onnx☆41Updated 9 months ago
- ☆23Updated 8 months ago
- Here we collect trick questions and failed tasks for open source LLMs to improve them.☆32Updated last year
- ☆33Updated 3 months ago
- TriNet: stabilizing self-supervised learning from complete or slow collapse on ASR.☆26Updated last year
- Run ONNX RWKV-v4 models with GPU acceleration using DirectML [Windows], or just on CPU [Windows AND Linux]; Limited to 430M model at this…☆20Updated last year
- A toolkit enhances PyTorch with specialized functions for low-bit quantized neural networks.☆28Updated 4 months ago
- RWKV model implementation☆38Updated last year
- [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling☆35Updated 11 months ago
- Inference RWKV with multiple supported backends.☆26Updated 2 months ago
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆20Updated this week
- A simple implementation of [Mamba: Linear-Time Sequence Modeling with Selective State Spaces](https://arxiv.org/abs/2312.00752)☆19Updated 9 months ago
- Trying to deconstruct RWKV in understandable terms☆14Updated last year
- BigKnow2022: Bringing Language Models Up to Speed☆14Updated last year
- Implementation of a modular, high-performance, and simplistic mamba for high-speed applications☆33Updated last week