TY  - RPRT
TI  - Orbax: Distributed Checkpointing with JAX
AU  - Colin Gaffney
AU  - Shutong Li
AU  - Daniel Ng
AU  - Anastasia Petrushkina
AU  - Niket Kumar
AU  - Adam Cogdell
AU  - Mridul Sahu
AU  - Yaning Liang
AU  - Nikhil Bansal
AU  - Justin Pan
AU  - Angel Mau
AU  - Abhishek Agrawal
AU  - Marco Berlot
AU  - Ruoxin Sang
AU  - Kiranbir Sodhia
AU  - Rakesh Iyer
PY  - 2026
UR  - https://arxiv.org/abs/2605.23066
ID  - 2605.23066
ER  -