google / orbax

git clone https://github.com/google/orbax.git

Orbax provides common checkpointing and persistence utilities for JAX users

—

stars, 24h

—

stars, 7 days

—

stars, 30 days

537

★ total

104

forks · 453 open issues

Star history starts 2026-10-20; the chart appears after the second daily snapshot.

LanguagePython

LicenseApache-2.0

Created2022-05-06 (4y ago)

Last pushtoday

OwnerOrganization

GitHub topics

checkpoint flax jax

Links

GitHub: github.com/google/orbax
Homepage: https://orbax.readthedocs.io/
Releases · Issues · Hacker News mentions