kingoflolz/mesh-transformer-jax - Gitstar Ranking

Gitstar Ranking

Users
Organizations
Repositories
Rankings

Sign in with GitHub

kingoflolz

Fetched on 2026/08/18 04:48

kingoflolz / mesh-transformer-jax

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

Star

6378

Rank

6159

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