minimaxir/mesh-transformer-jax - Gitstar Ranking

Gitstar Ranking

Users
Organizations
Repositories
Rankings

Sign in with GitHub

minimaxir

Fetched on 2026/08/18 02:27

minimaxir / mesh-transformer-jax

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

Star

3

Rank

3461565

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