bot_experiments/IBM_Transformer+TimeEmbedding.ipynb training very slow? I'm just wondering if there is any way to speed it?