flax

d9e6e44b - Use gfile.listdir instead of gfile.glob for checkpointing. The former is proven to be much more efficient in GCS.