flax

7974b106 - Merge pull request #4580 from google:nnx-tabulate-issue