GPU-Puzzles

收录理由

训练模型的人大多从未亲手写过一行 GPU 代码，而这个交互式笔记本正是弥补这一缺口最直接的途径之一。谜题由浅入深，从最简单的向量映射开始，逐步推进到共享内存与归约算法——这些正是当今多数深度学习框架背后的核心运算。代码基于 Numba 运行，你写的看似 Python 的函数会被编译成真正的 CUDA 内核，入门门槛低，但硬件层面发生的事情并未被掩盖。每个谜题都会将你写的内核与参考实现做比对，索引是否正确立刻见分晓。最理想的使用方式是把这当作一次动手练习：在 Colab 中打开它，开启 GPU 模式，然后直接上手解题，而不是只读文字说明。对于任何刚接触 GPU 编程的人来说，这也是一份扎实的开篇练习，并且与 Sasha Rush 配套的 Tensor Puzzles（针对 PyTorch）自然衔接。