kingoflolz / mesh-transformer-jax

Model parallel transformers in JAX and Haiku
6,334Updated 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: