GitHub repository: kingoflolz/mesh-transformer-jax
Primary language: Python
Created: 2021-03-13T23:31:14Z
Description: Model parallel transformers in JAX and Haiku