Triton-Puzzles and Tensor-Puzzles
These are complementary learning resources—Triton-Puzzles teaches the Triton compiler for GPU kernels while Tensor-Puzzles teaches PyTorch fundamentals, so learners typically progress from PyTorch understanding to optimized Triton kernel implementation.
About Triton-Puzzles
gpu-mode/Triton-Puzzles
Puzzles for learning Triton
About Tensor-Puzzles
srush/Tensor-Puzzles
Solve puzzles. Improve your pytorch.
Comprised of 21 self-contained puzzles that isolate specific tensor operations, each solvable in a single line using only broadcasting, indexing, and arithmetic—no standard library functions allowed. Emphasizes mastery of NumPy/PyTorch's broadcasting semantics through constraint-based learning, with visual diagram feedback and property-based testing via Hypothesis. Targets PyTorch and NumPy users seeking deeper understanding of low-level tensor manipulation fundamentals.
Scores updated daily from GitHub, PyPI, and npm data. How scores work