Open source repositories tagged with #checkpoint, ranked by health score.
Orbax provides common checkpointing and persistence utilities for JAX users
