@misc{indiciaef1df1b46a9f9,
  title = {Scalable Training of Language Models using JAX pjit and TPUv4},
  author = {Joanna Yoo and Kuba Perlin and Siddhartha Rao Kamalakara and João G. M. Araújo},
  year = {2022},
  url = {https://arxiv.org/abs/2204.06514},
  note = {Source identifier: 2204.06514}
}