flax

df98b59b - Merge branches 'master' and 'master' of http://github.com/google/flax