Implementation of GateLoop Transformer in Pytorch and Jax - View it on GitHub