flax

b5f478ff - Merge pull request #4536 from mattbahr:accuracy-metric-binary-classification-support