title: How to convert tf.Keras model to TPU using TensorFlow 2.0 in Google Colab?
body: <p>Since TF 2.0 doesn't have the tf.contrib layer, how do I convert my model to run the training on TPU, without having access to <code>tf.contrib.tpu.keras_to_tpu_model()</code>?</p>
  
  <p>I tried looking for code, but all of it are running on TensorFlow 1.x</p>
  
  <p>My data is in <code>.npy</code>, I have a simple model, and I'm using only <code>model.compile()</code> and <code>model.fit()</code> to train it, but it looks like the model is running on CPU (taking 30 minutes/epoch vs the 2 minutes/epoch on GPU).</p>