Orbax

https://github.com/google/orbax

Orbax provides common checkpointing and persistence utilities for JAX users, offering a customizable...

Tokens:62,746
Snippets:783
Trust Score:8.9
License:Apache-2.0
Update:3 weeks ago
Tokens:
Raw