flax

c289bf11 - Move latest_checkpoint functionality to its own method, similar to tf.train.latest_checkpoint.