flax

8025b3dd - Merge pull request #4623 from mattbahr:implement-spectral-and-instance-norm