kingoflolz / mesh-transformer-jaxLinks

Model parallel transformers in JAX and Haiku
6,333Updated 2 years ago

Alternatives and similar repositories for mesh-transformer-jax

Users that are interested in mesh-transformer-jax are comparing it to the libraries listed below

Sorting: