flax

d883f3e8 - Refactor cifar10 to accept only FLAG values and move validations as part of FLAG definitions.