Model parallel transformers in JAX and Haiku
Primary LanguagePython
No issues in this repository yet.