flax

c4de032f - Use tf.io.gfile instead of adding an explicit import for gfile. This is consistent with the examples/