Model parallel transformers in JAX and Haiku - View it on GitHub
Star
0
Rank
11230011