Orbax

Software / App

A library within the Jax AI stack for checkpointing, designed to handle distributed environments.

Mentioned in 1 video

Videos Mentioning Orbax

AI Dev 25 x NYC | Robert Crowe: JAX Made Simple: An Intuitive Guide to Building Fast Neural Networks

AI Dev 25 x NYC | Robert Crowe: JAX Made Simple: An Intuitive Guide to Building Fast Neural Networks

DeepLearningAI

A library within the Jax AI stack for checkpointing, designed to handle distributed environments.