Fixed typos in the docstrings using antigravity#5145

vfdev-5 wants to merge 1 commit intomaingoogle/flax:mainfrom

fix-more-typosgoogle/flax:fix-more-typosCopy head branch name to clipboard