Figure out why math isn't working

E.g. https://opensource.posit.co/blog/ai/keydanatorchconvlstm/

<img width="388" height="165" alt="Image" src="https://github.com/user-attachments/assets/87a4c269-6778-46c6-a006-2f1b0d5f19a7" />