google/orbax

Orbax provides common checkpointing and persistence utilities for JAX users

↗ Open on GitHub

Stars

534

Forks

101

Language

Python

License

Apache-2.0

Last push

16 hours ago

Python jax checkpoint flax

No related intel yet

This repo has not appeared in any of the sources the radar tracks. The collector runs on a schedule — check back once it covers this repo.