flax

960eb6d0 - Merge branch 'master' of http://github.com/google/flax into automasking