Thanks so much @KFrank, that sounds like the right advice.
One question, though. I understand that the weight updating step should be different from the precomputing inputs step, but I thought forward was the model entry point. How can I separate the two in practice?
— Arnav