kingoflolz/mesh-transformer-jax - Gitstar Ranking

Gitstar Ranking

Users
Organizations
Repositories
Rankings

Sign in with GitHub

kingoflolz

Fetched on 2026/06/08 18:51

kingoflolz / mesh-transformer-jax

Model parallel transformers in JAX and Haiku - View it on GitHub

Star

6374

Rank

5974

Released by @k0kubun in December 2014. Fork me on GitHub.