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