Tinygrad: Implement an LSTM Cell from ScratchDeep-ML | Problem