minimaxir/mesh-transformer-jax - Gitstar Ranking

Gitstar Ranking

Users
Organizations
Repositories
Rankings

Sign in with GitHub

minimaxir

Fetched on 2026/07/13 17:13

minimaxir / mesh-transformer-jax

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

Star

3

Rank

3389669

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