GitHub repository: samos123/orbax
Description: Orbax provides common checkpointing and persistence utilities for JAX users