search

google / orbax

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

[Experimental Feature] Support `empty NamedTuple` leaf when `PyTreeMetadataOptions.support_rich_types=true`. #1362

Open copybara-service[bot] opened 1 day ago

copybara-service[bot] commented 1 day ago

[Experimental Feature] Support empty NamedTuple leaf when PyTreeMetadataOptions.support_rich_types=true.