flax

c2c3feaa - Merge pull request #3510 from carlosgmartin:normalization_mask