search

google / orbax

Orbax provides common checkpointing and persistence utilities for JAX users
https://orbax.readthedocs.io/
Apache License 2.0
307 stars 36 forks source link

Refactor `*pytree*handler.py` to improve readability of metadata/tree api usage. #1315

Closed copybara-service[bot] closed 2 weeks ago

copybara-service[bot] commented 2 weeks ago

Refactor *pytree*handler.py to improve readability of metadata/tree api usage.