Implement an LSTM from Scratch in PyTorch

Mediumauto-graded

Code the LSTM cell equations from scratch: forget gate, input gate, cell state update, and output gate using raw tensors.

Solve it

Open in ColabQuestion notebookSolution

Check your answer

The grader verifies properties of your implementation, so a correct solution written differently from ours still passes.

pip install torchleet

from torchleet import check
check("lstm", CustomLSTMModel, LSTMModel)