tensorpro / tpu_rwkv
JAX implementations of RWKV
☆19Updated last year
Alternatives and similar repositories for tpu_rwkv:
Users that are interested in tpu_rwkv are comparing it to the libraries listed below
- Trying to deconstruct RWKV in understandable terms☆14Updated last year
- ☆42Updated 2 years ago
- A converter and basic tester for rwkv onnx☆42Updated last year
- RWKV model implementation☆37Updated last year
- tinygrad port of the RWKV large language model.☆44Updated last month
- Interpretability analysis of language model outlier and attempts to distill the model☆13Updated 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 2 years ago
- This project aims to make RWKV Accessible to everyone using a Hugging Face like interface, while keeping it close to the R and D RWKV bra…☆64Updated last year
- Training a reward model for RLHF using RWKV.☆14Updated last year
- RWKV, in easy to read code☆71Updated 3 weeks ago
- 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
- Framework agnostic python runtime for RWKV models☆146Updated last year
- RWKV-v2-RNN trained on the Pile. See https://github.com/BlinkDL/RWKV-LM for details.☆67Updated 2 years ago
- Course Project for COMP4471 on RWKV☆17Updated last year
- Code for paper: "QuIP: 2-Bit Quantization of Large Language Models With Guarantees" adapted for Llama models☆36Updated last year
- https://x.com/BlinkDL_AI/status/1884768989743882276☆27Updated 2 months ago
- Fast modular code to create and train cutting edge LLMs☆66Updated 10 months ago
- A large-scale RWKV v6, v7(World, ARWKV, PRWKV) inference. Capable of inference by combining multiple states(Pseudo MoE). Easy to deploy o…☆34Updated 3 weeks ago
- Demonstration that finetuning RoPE model on larger sequences than the pre-trained model adapts the model context limit☆63Updated last year
- A library for simplifying fine tuning with multi gpu setups in the Huggingface ecosystem.☆16Updated 5 months ago
- Experiments with BitNet inference on CPU☆53Updated last year
- Experiments on the impact of depth in transformers and SSMs.☆24Updated 5 months ago
- Zeus LLM Trainer is a rewrite of Stanford Alpaca aiming to be the trainer for all Large Language Models☆69Updated last year
- Chatbot that answers frequently asked questions in French, English, and Tunisian using the Rasa NLU framework and RWKV-4-Raven☆13Updated last year
- GoldFinch and other hybrid transformer components☆45Updated 8 months ago
- ☆40Updated 2 years ago
- Code for the paper: https://arxiv.org/pdf/2309.06979.pdf☆19Updated 8 months ago
- If it quacks like a tensor...☆58Updated 5 months ago
- ☆13Updated 4 months ago
- Implementation of Spectral State Space Models☆16Updated last year