flax

22ed17e1 - Merge pull request #3805 from google:nnx-simplify-readme-again-2