Nosotros tenemos ya una RNN implementada en nuestra libreria. En este notebook hay un ejemplo de como usarla: https://github.com/Nixtla/neuralforecast/blob/main/nbs/models_rnn__rnn.ipynb. Podrias probar de usar nuestra implementacion?