flax

4ba62991 - Merge pull request #3864 from google:nnx-static-issue