flax

Take 2: Add .style.yapf config that matches the Google internal style guide.

#400

Merged