flax

9541acbd - Merge pull request #3707 from google:nnx-simplify-readme