tensorpro / tpu_rwkvLinks
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
Sorting:
- Trying to deconstruct RWKV in understandable terms☆14Updated 2 years ago
- RWKV model implementation☆38Updated 2 years ago
- Interpretability analysis of language model outlier and attempts to distill the model☆13Updated 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…☆65Updated 2 years ago
- A converter and basic tester for rwkv onnx☆43Updated last year
- Framework agnostic python runtime for RWKV models☆146Updated 2 years ago
- tinygrad port of the RWKV large language model.☆45Updated 5 months 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
- ☆43Updated 2 years ago
- RWKV, in easy to read code☆71Updated 5 months ago
- Experiments with BitNet inference on CPU☆54Updated last year
- ☆40Updated 2 years ago
- ☆49Updated last year
- Training a reward model for RLHF using RWKV.☆15Updated 2 years ago
- Run ONNX RWKV-v4 models with GPU acceleration using DirectML [Windows], or just on CPU [Windows AND Linux]; Limited to 430M model at this…☆21Updated 2 years ago
- RWKV in nanoGPT style☆192Updated last year
- Prepare for DeekSeek R1 inference: Benchmark CPU, DRAM, SSD, iGPU, GPU, ... with efficient code.☆73Updated 6 months ago
- A torchless, c++ rwkv implementation using 8bit quantization, written in cuda/hip/vulkan for maximum compatibility and minimum dependenci…☆313Updated last year
- Demonstration that finetuning RoPE model on larger sequences than the pre-trained model adapts the model context limit☆63Updated 2 years ago
- GoldFinch and other hybrid transformer components☆46Updated last year
- Advanced Ultra-Low Bitrate Compression Techniques for the LLaMA Family of LLMs☆110Updated last year
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆59Updated 3 years ago
- Course Project for COMP4471 on RWKV☆17Updated last year
- Token Omission Via Attention☆128Updated 10 months ago
- ☆39Updated 2 years ago
- ☆28Updated last year
- My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated☆33Updated last year
- Fast modular code to create and train cutting edge LLMs☆68Updated last year
- SparseGPT + GPTQ Compression of LLMs like LLaMa, OPT, Pythia☆41Updated 2 years ago
- BlinkDL's RWKV-v4 running in the browser☆47Updated 2 years ago