GitHub repository: lucidrains/gateloop-transformer
Primary language: Python
Description: Implementation of GateLoop Transformer in Pytorch and Jax