tf.keras on TPUs on Colab – TensorFlow – Medium

Did you know that Colab includes the ability to select a free Cloud TPU for training models? That's right, a whole TPU for you to use all by yourself in a notebook! As of TensorFlow 1.11, you can train Keras models with TPUs. In this post, let's take a look at what changes you need to make to your code to be able to train a Keras model on TPUs. Note that some of this may be simplified even further with the release of TensorFlow 2.0 later this year, but I thought it'd be helpful to share these tips in case you'd like to try this out now.