Giters

lawrendran / mesh-transformer-jax

Model parallel transformers in JAX and Haiku

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

lawrendran/mesh-transformer-jax Stargazers