flax

13131938 - Merge pull request #5142 from google:docs-typo-fix-_cached_partial