transformers

f82a2a5e - [Benchmark] Add benchmarks for TF Training (#5594)

[Benchmark] Add benchmarks for TF Training (#5594) * tf_train * adapt timing for tpu * fix timing * fix timing * fix timing * fix timing * update notebook * add tests