ym-han / mesh-transformer-jax

Model parallel transformers in JAX and Haiku

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ym-han/mesh-transformer-jax Issues

No issues in this repository yet.