flax

7330d6fc - Merge pull request #3689 from chiamp:nnx_norm