TPU
Google ColaboratoryのTPUを試してみる - Qiita
Google ColabでKerasからTPUを使う方法とGPUとの速度差 - モーグルとカバとパウダーの日記
TPUに切り替えるだけではダメ、モデルを変換する
predictの時にはCPUモデルに戻す
Google ColabのTPUでtransformerを学習 - 言語と知能