flax
d1935c71 - Make FlatState a Mapping instead of a dict

Commit

2 years ago

Make FlatState a Mapping instead of a dict Also, make traverse_util.flatten_dict accept mappings. Fixes https://github.com/google/flax/issues/3879

References

Author

NeilGirdhar

Committer

NeilGirdhar

Parents