@misc{indiciae2770ca6d4a0b,
  title = {Orbax: Distributed Checkpointing with JAX},
  author = {Colin Gaffney and Shutong Li and Daniel Ng and Anastasia Petrushkina and Niket Kumar and Adam Cogdell and Mridul Sahu and Yaning Liang and Nikhil Bansal and Justin Pan and Angel Mau and Abhishek Agrawal and Marco Berlot and Ruoxin Sang and Kiranbir Sodhia and Rakesh Iyer},
  year = {2026},
  url = {https://arxiv.org/abs/2605.23066},
  note = {Source identifier: 2605.23066}
}