/nanodl

A Jax-based library for designing and training transformer models from scratch.

Primary LanguagePythonMIT LicenseMIT

Stargazers