GitHub repository: simudt/Griffin-Jax
Primary language: Python
Created: 2024-03-06T07:01:34Z
Description: Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"