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