flax

a2fe3ca2 - Merge pull request #4745 from vfdev-5:fix-lm1b-nnx-example