Training seq2seq model on Google Colab TPU with big dataset - Keras

Question

I\'m trying to train a sequence to sequence model for machine translation using Keras on Google Colab TPU. I have a dataset which I can load in memory but I have to preprocess t