TY  - RPRT
TI  - Scalable Training of Language Models using JAX pjit and TPUv4
AU  - Joanna Yoo
AU  - Kuba Perlin
AU  - Siddhartha Rao Kamalakara
AU  - João G. M. Araújo
PY  - 2022
UR  - https://arxiv.org/abs/2204.06514
ID  - 2204.06514
ER  -