Pinned Repositories
hackathon-tardis
git-sandbox
mesh-transformer-jax
Model parallel transformers in JAX and Haiku
nooney's Repositories
nooney/git-sandbox
nooney/mesh-transformer-jax
Model parallel transformers in JAX and Haiku
Model parallel transformers in JAX and Haiku
Model parallel transformers in JAX and Haiku