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
- 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
- RWKV, in easy to read code☆67Updated 2 months 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
- A converter and basic tester for rwkv onnx☆42Updated last year
- tinygrad port of the RWKV large language model.☆44Updated 8 months 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
- Training a reward model for RLHF using RWKV.☆14Updated last year
- RWKV model implementation☆37Updated last year
- Interpretability analysis of language model outlier and attempts to distill the model☆13Updated last year
- Framework agnostic python runtime for RWKV models☆145Updated last year
- ☆11Updated last year
- RWKV-7: Surpassing GPT☆79Updated 3 months ago
- Course Project for COMP4471 on RWKV☆17Updated last year
- My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated☆31Updated 6 months ago
- GoldFinch and other hybrid transformer components☆43Updated 7 months ago
- RWKV in nanoGPT style☆187Updated 8 months ago
- Fast modular code to create and train cutting edge LLMs☆65Updated 9 months ago
- SparseGPT + GPTQ Compression of LLMs like LLaMa, OPT, Pythia☆41Updated last year
- ☆20Updated last year
- Experiments with BitNet inference on CPU☆53Updated 10 months ago
- FlexAttention w/ FlashAttention3 Support☆26Updated 4 months ago
- Latent Large Language Models☆17Updated 5 months ago
- [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling☆35Updated last year
- A library for simplifying fine tuning with multi gpu setups in the Huggingface ecosystem.☆16Updated 3 months ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆44Updated 8 months ago