flax

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

#934

Merged