flax

46126954 - Merge pull request #1574 from google:avital-prefetch-doc