Solved by JamesWright in post #4 
              
              
                I’ve tested it against TensorFlow/Keras outputs, it does not work (output is different). I’ve found following conversion to work for GRU layer (TensorFlow r2.4rc0 - tf.keras.layers.GRU to PyTorch 1.9.0 - torch.nn.GRU): 

Save weights from TensorFlow model (for example to .npz file):

import random
i…