kingoflolz/mesh-transformer-jax: Model parallel transformers in JAX and Haiku
Category: nlp
Language: Python