Last released Oct 29, 2024
Orbax Checkpoint
Last released Jul 1, 2024
Orbax Export
Last released Apr 10, 2024
Set of libraries for ML training loops in JAX.
Last released Sep 6, 2023
Orbax
Supported by