flax

36bbb414 - Merge pull request #3385 from chiamp:early_stopping