Cover Image

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!

Similar Items

A Multi-agent AI System for Deep Learning Model Migration from TensorFlow to JAX
by: Nikolov, Stoyan, et al.
Published: (2026)

How Financial Planning Helps Private Sector Employees Grow Sustainably: Insights from Bilaspur Division
by: Agrawal, Rajesh, et al.
Published: (2025)

School Library Media Management Skills: A Few Tips.
by: Cogdell, Edna
Published: (2003)

A Polynomial Kernel for Proper Helly Circular-arc Vertex Deletion
by: Agrawal, Akanksha, et al.
Published: (2024)

Mechanistic Insights into Dimerization and Cross-β Sheet Formation in the HIV-Associated Amyloidogenic Peptide PAP248–286 from Unbiased All-Atom Molecular Dynamics Simulations
by: Agrawal, Nikhil
Published: (2025)

A mark and recapture perspective on vaccination touchpoints
by: Thakkar, Niket
Published: (2025)

Enterprise Digital Brain An AI-Augmented System for Knowledge Organization and Cognitive Productivity
by: Agrawal, Rakesh Kumar
Published: (2026)

Smart ICU Monitoring System Using IoT and Edge AI for Real-Time Patient Risk Prediction
by: Agrawal, Rakesh Kumar
Published: (2026)

The Economic Impact of Low- and High-Frequency Temperature Changes
by: Gospodinov, Nikolay, et al.
Published: (2025)

BlackJAX: Composable Bayesian inference in JAX
by: Cabezas, Alberto, et al.
Published: (2024)

Rethinking Immune Checkpoint Inhibition in Cancer
by: Tibor Bakacs, et al.
Published: (2026)

Attribute Diversity Determines the Systematicity Gap in VQA
by: Berlot-Attwell, Ian, et al.
Published: (2023)

Consolidation of multiple binary distillation columns for large heat duty savings
by: Parikshit S. Kadu, et al.
Published: (2026)

An Improved Bound for the Beck-Fiala Conjecture
by: Bansal, Nikhil, et al.
Published: (2025)

Decoupling via Affine Spectral-Independence: Beck-Fiala and Komlós Bounds Beyond Banaszczyk
by: Bansal, Nikhil, et al.
Published: (2025)

Quasi-Monte Carlo Beyond Hardy-Krause
by: Bansal, Nikhil, et al.
Published: (2024)

Investigation of Structural, Mechanical and Electrochemical Properties of Temperature‐Dependent Dual AlN/TiAlN Coating Synthesized by CVD Process
by: Soham Das, et al.
Published: (2026)

Volume 1
by: Gaffney, Vincent
Published: (2022)

Humic and Fulvic Acids : isolation, structure, and environmental role / Jeffrey S. Gaffney, Nancy A. Marley and Sue B. Clark
by: S. Gaffney J
Published: (1996)

Interlibrary Loan Requests for Locally Available Materials: WorldCat Local's Impact
by: Gaffney, Megan
Published: (2012)

LLM Library Learning Fails: A LEGO-Prover Case Study
by: Berlot-Attwell, Ian, et al.
Published: (2025)

Library Learning Doesn't: The Curious Case of the Single-Use "Library"
by: Berlot-Attwell, Ian, et al.
Published: (2024)

DrJAX: Scalable and Differentiable MapReduce Primitives in JAX
by: Rush, Keith, et al.
Published: (2024)

DAC-JAX: A JAX Implementation of the Descript Audio Codec
by: Braun, David
Published: (2024)

USA Tariffs Effect: Machine Learning Insights into the Stock Market
by: Patel, Mridul
Published: (2025)

A Generic Method to Extend Message Space of a Strong Pseudorandom Permutation
by: Mridul Nandi
Published: (2009)

Duti Kobita (Two Poems)
by: Mridul Dasgupta
Published: (2019)

Local Transfer and Reducibility of Induced Representations of $p$-adic Groups of Classical Type
by: Asgari, Mahdi, et al.
Published: (2014)

Rankin-Selberg L-Functions for GSpin x GL Groups
by: Asgari, Mahdi, et al.
Published: (2024)

Transitioning from Curriculum Materials Center to School Library Media Center in Pre-Service Teacher Education
by: Dickinson, Gail, et al.
Published: (2004)

Task Priors: Enhancing Model Evaluation by Considering the Entire Space of Downstream Tasks
by: Patel, Niket, et al.
Published: (2025)

On the Local Complexity of Linear Regions in Deep ReLU Networks
by: Patel, Niket, et al.
Published: (2024)

GODMAX: Modeling gas thermodynamics and matter distribution using JAX
by: Pandey, Shivam, et al.
Published: (2024)

Modeling Dynamic (De)Allocations of Local Memory for Translation Validation
by: Rose, Abhishek, et al.
Published: (2024)

Quantum Machine Learning: Unveiling Trends, Impacts through Bibliometric Analysis
by: Bansal, Riya, et al.
Published: (2025)

Quantum State Preparation for Medical Data: Comprehensive Methods, Implementation Challenges, and Clinical Prospects
by: Rajput, Nikhil Kumar, et al.
Published: (2025)

Scrutinizing Fermionic Dark Matter in Scotogenic Model with Low Reheating Temperature
by: Roy, Abhishek, et al.
Published: (2025)

DART: aDaptive Accept RejecT for non-linear top-K subset identification
by: Agarwal, Mridul, et al.
Published: (2020)

Coulomb-Corrected Wormhole Model for Neon-20
by: Bai, Ruoxin, et al.
Published: (2025)

“Parir no es un asunto de etnia, es un asunto de humanidad”: experiencias frente a la violencia obstétrica durante la atención al parto en mujeres indígenas
by: Emily Gaffney Gleason
Published: (2021)