Colab notebooks

TensorFlow Colab notebooks

• Fashion MNIST with Keras and TPU

  This notebook demonstrates an end-to-end image classification sample with data loading, TPU training, model export, and deployment.

• Shakespeare with Keras and TPU

  This notebook uses Keras to build a language model and train it on a Cloud TPU. This language model predicts the next character of text given the text so far. The trained model can generate new snippets of text that read in a similar style to the text training data.

PyTorch Colab notebooks

• PyTorch/TPU MNIST Demo

  This colab example corresponds to the implementation under test_train_mnist.py and is TF/XRT 1.15 compatible.

• PyTorch/TPU ResNet18/CIFAR10 Demo

  This colab example corresponds to the implementation under test_train_cifar.py and is TF/XRT 1.15 compatible.

• PyTorch/TPU ResNet50 Inference Demo

  This colab example is TF/XRT 1.15 compatible.