@article{scalabletrainingoflanguagemodelsusing,
  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},
  eprint = {2204.06514},
  archivePrefix = {arXiv},
  url = {https://arxiv.org/abs/2204.06514v1},
}