Last released May 26, 2024
A Jax-based library for designing and training transformer models from scratch.
Supported by