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
- ☆42Updated last year
- Trying to deconstruct RWKV in understandable terms☆14Updated last year
- RWKV, in easy to read code☆71Updated 3 months ago
- Course Project for COMP4471 on RWKV☆17Updated last year
- tinygrad port of the RWKV large language model.☆44Updated 2 weeks ago
- RWKV model implementation☆37Updated last year
- Interpretability analysis of language model outlier and attempts to distill the model☆13Updated last year
- Training a reward model for RLHF using RWKV.☆14Updated last year
- 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
- A converter and basic tester for rwkv onnx☆42Updated 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
- Let us make Psychohistory (as in Asimov) a reality, and accessible to everyone. Useful for LLM grounding and games / fiction / business /…☆40Updated last year
- ☆11Updated last year
- My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated☆31Updated 7 months ago
- Framework agnostic python runtime for RWKV models☆145Updated last year
- Code for paper: "QuIP: 2-Bit Quantization of Large Language Models With Guarantees" adapted for Llama models☆36Updated last year
- GoldFinch and other hybrid transformer components☆45Updated 8 months ago
- Demonstration that finetuning RoPE model on larger sequences than the pre-trained model adapts the model context limit☆63Updated last year
- RWKV in nanoGPT style☆187Updated 9 months ago
- SparseGPT + GPTQ Compression of LLMs like LLaMa, OPT, Pythia☆41Updated 2 years ago
- QuIP quantization☆52Updated last year
- FlexAttention w/ FlashAttention3 Support☆26Updated 5 months ago
- RWKV-v2-RNN trained on the Pile. See https://github.com/BlinkDL/RWKV-LM for details.☆67Updated 2 years ago
- Experiments on the impact of depth in transformers and SSMs.☆23Updated 4 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…☆32Updated last week
- BigKnow2022: Bringing Language Models Up to Speed☆14Updated last year