By: hardik (@TheHardikVala)
I'm a huge fan of Sasha Rush's Tensor Puzzles for training your tensor programming muscle. LLM's are pretty good at writing pytorch or numpy code, but you still need a solid grasp of tensor operations to verify the code, and the code they write isn't expert quality (yet).
This notebook is a sequel to Tensor Puzzles, for those looking for additional practice, with a whole different collection of problems.
I highly recommend running the notebook in Colab. Click here and copy the notebook to start.
All diagrams were generated with the help of tensordiagram—a special-purpose python library for visualizing tensors—and chalk—a general-purpose python graphics library.