flax
2d70c866 - Save types information to allow for fully self-describing checkpoints. Add type-specific metadata to `LazyValue` so that more information can be gleaned without fully restoring the tree.

Commit

2 years ago

Save types information to allow for fully self-describing checkpoints. Add type-specific metadata to `LazyValue` so that more information can be gleaned without fully restoring the tree. PiperOrigin-RevId: 537384891

References

test_537384891

#3136 - Save types information to allow for fully self-describing checkpoints. Add type-specific metadata to `LazyValue` so that more information can be gleaned without fully restoring the tree.

Author

cpgaffney1

Committer

a-googler

Parents

266fd52d

flax 2d70c866 - Save types information to allow for fully self-describing checkpoints. Add type-specific metadata to `LazyValue` so that more information can be gleaned without fully restoring the tree.

flax
2d70c866 - Save types information to allow for fully self-describing checkpoints. Add type-specific metadata to `LazyValue` so that more information can be gleaned without fully restoring the tree.