flax

48f3a704 - Merge pull request #3646 from google:compact_name_scope_v3