I am training an LSTM model (the architecture is very simple). For whatever reason, and randomly, usually after 15 or so epochs, i get this message that the kernel has died and needs to restart.

So this isn't due to inactivity

I have an R4 instance type.

What is the problem?

And when will Tensorflow 2.4 be available?