I'm making a visual explainer site for PyTorch functions:

https://whytorch.org/

A great example of how it works is http://whytorch.org/torch.amax/

Clicking items in the tensors explains where they came from and where they are used in the output. The input tensors can be modified too.

It's a one-man side project that's been half building the site framework, and half re-implementing pytorch functions in javascript. Plenty more functions to go, but hopefully people can already find it useful. I'm planning on doing a Show HN once I've added ~10 more functions.

Posting this from a throwaway account because my main account is locked due to `noprocrast`!

This looks neat. Great idea!

Perhaps `dot`, `cross` etc might be useful additions.