I made these a couple of years ago as a teaching exercise for https://minitorch.github.io/ . At the time the resources for doing anything on GPUs were pretty sparse and the NVidia docs were quite challenging. These days there are great resources for going deep on this topic. The CUDA-mode org is particularly great, both their video series and PMPP reading groups.
Thanks a lot, Sasha, for creating these. I found your LLM training puzzles to be excellent as well.
https://github.com/srush/Triton-puzzles https://github.com/srush/tensor-puzzles https://github.com/srush/autodiff-puzzles https://github.com/srush/transformer-puzzles https://github.com/srush/GPTworld https://github.com/srush/LLM-Training-Puzzles