dtunai / Griffin-JaxView external linksLinks
Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"
☆15May 10, 2024Updated last year
Alternatives and similar repositories for Griffin-Jax
Users that are interested in Griffin-Jax are comparing it to the libraries listed below
Sorting:
- Implementation of Griffin from the paper: "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆56Oct 27, 2025Updated 3 months ago
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆67Apr 24, 2024Updated last year
- ☆51Jan 28, 2024Updated 2 years ago
- ☆14Jul 12, 2022Updated 3 years ago
- ☆29Feb 27, 2024Updated last year
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Jan 7, 2025Updated last year
- Implementation of Spectral State Space Models☆16Feb 23, 2024Updated last year
- ☆106Mar 9, 2024Updated last year
- Blog post☆17Feb 16, 2024Updated 2 years ago
- ☆15May 30, 2024Updated last year
- HGRN2: Gated Linear RNNs with State Expansion☆56Aug 20, 2024Updated last year
- ☆24Sep 25, 2024Updated last year
- Here we will test various linear attention designs.☆62Apr 25, 2024Updated last year
- [CoLM 24] Official Repository of MambaByte: Token-free Selective State Space Model☆24Oct 12, 2024Updated last year
- Bagz is a format for storing a sequence of string records. It supports per-record compression and fast index-based lookup.☆51Jul 22, 2025Updated 6 months ago
- An annotated implementation of the Hyena Hierarchy paper☆34May 28, 2023Updated 2 years ago
- ☆35Feb 26, 2024Updated last year
- Structured matrices for compressing neural networks☆67Oct 5, 2023Updated 2 years ago
- Code repository for "Multi-Task Encoder-Dual-Decoder Modeling Framework on Mixed Frequency Data", International Journal of Forecasting, 2…☆12Feb 18, 2024Updated last year
- ☆44Nov 1, 2025Updated 3 months ago
- ☆35Nov 22, 2024Updated last year
- Official implementation of the paper "Pretraining Language Models to Ponder in Continuous Space"☆24Jul 21, 2025Updated 6 months ago
- Fast, free, easy, and object-agnostic video anonymization☆11Dec 12, 2020Updated 5 years ago
- Bayesian adaptive stimulus placement of psychometric function for MATLAB.☆10Nov 7, 2018Updated 7 years ago
- Griffin MQA + Hawk Linear RNN Hybrid☆89Apr 26, 2024Updated last year
- A nonparametric variational information bottleneck (NVIB) layer in Pytorch☆11Apr 15, 2025Updated 10 months ago
- ☆39Apr 5, 2024Updated last year
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆98Dec 5, 2024Updated last year
- Official implementation of Privacy Implications of Retrieval-Based Language Models (EMNLP 2023). https://arxiv.org/abs/2305.14888☆37Jun 10, 2024Updated last year
- ☆40Jan 5, 2024Updated 2 years ago
- Pytorch-like dataloaders for JAX.☆101Dec 16, 2025Updated 2 months ago
- Virtual Bash interpreter with a virtual file system for multi-tenant environments.☆31Updated this week
- Generating Potent Poisons and Backdoors from Scratch with Guided Diffusion☆11Apr 1, 2024Updated last year
- ☆11Oct 6, 2020Updated 5 years ago
- Generate v4 UUIDs using libsodium's RNG☆11Jun 16, 2020Updated 5 years ago
- Wenet speech to text for react native☆10Nov 1, 2022Updated 3 years ago
- Node.js Logical reasoning machine (WIP)☆10Dec 18, 2014Updated 11 years ago
- Generate a menu with selectable menu items as a string☆12Dec 26, 2018Updated 7 years ago
- PCM audio sample rate conversion for Node.js☆15May 13, 2013Updated 12 years ago