mesh-transformer-jax

Python ★ 0 updated 4y ago ⑂ fork

Model parallel transformers in JAX and Haiku

No plain-English explanation yet — one is being written right now. Check back in a minute.