Nice to see it!
I think that a lot of mathematics would benefit from executable code (or I would say - executable maths) - so people can play and experiment with it.
I had my own approach for teaching deep learning (https://github.com/stared/thinking-in-tensors-writing-in-pyt...), showing both mathematical formulation in LaTex, and a numeric one in PyTorch.
would be great if upgrades to pytorch 2
In case I use it for another course or training, I will surely do.