flax

61ece402 - Merge pull request #3652 from chiamp:instancenorm