Orbax: Distributed Checkpointing with JAX

Fuente: arXiv

Saved in:

Bibliographic Details
Main Authors:	Gaffney, Colin, Li, Shutong, Ng, Daniel, Petrushkina, Anastasia, Kumar, Niket, Cogdell, Adam, Sahu, Mridul, Liang, Yaning, Bansal, Nikhil, Pan, Justin, Mau, Angel, Agrawal, Abhishek, Berlot, Marco, Sang, Ruoxin, Sodhia, Kiranbir, Iyer, Rakesh
Format:	Preprint
Published:	2026
Subjects:	Distributed, Parallel, and Cluster Computing Machine Learning
Online Access:	Acceder al recurso
Tags:	Add Tag No Tags, Be the first to tag this record!

Internet

https://arxiv.org/abs/2605.23066