flax

c86c8b92 - No label smoothing for evaluation and fixes a typo in input_pipeline.