If you like these puzzles, you might also enjoy the minitorch [1] teaching library by the same author, which is a course where you implement a simplified version of the pytorch library from scratch.

Completing that course and understanding the differences between the simplified and full versions has been the most useful deep learning teaching resource for me to date.

[1] https://minitorch.github.io/