Just as a side note Keras LSTM and PyTorch LSTM are different and not compatible 100%. It would be good to see side by side analysis of the difference, but I cannot provide that since I don’t know for such document.