google/orbax

**Brief Summary** Orbax is a Google‑maintained Python library that supplies reusable checkpointing and persistence utilities for JAX‑based machine‑learning pipe