Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
StableVery rare

kingoflolz/mesh-transformer-jax

Model parallel transformers in JAX and Haiku

Owner
kingoflolz (personal account)
Stars
Stars: 6,390
Forks
Forks: 876
Language
Python
License
Apache-2.0
Created
2021
Tier
Stable (5,000 to 19,999 ★) What this tier means
View on GitHub

More repos by kingoflolz

All 2 repos by kingoflolz

Nearby in Python

All Python repos