Adding call back to measure time of each step 
Hi
I need to measure the time of each training step, with distributed version of seq2seq_trainer with TPUs, could you provide me with some hints how to measure the time? thanks 
This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.
