Giters

ashokpant / mesh-transformer-jax

Model parallel transformers in JAX and Haiku

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ashokpant/mesh-transformer-jax Stargazers