Train very large language models in Jax.
Primary LanguagePythonMIT LicenseMIT
No issues in this repository yet.