flax

2bf5748d - Merge pull request #4912 from google:no-copy-merge